requirements.txt

accelerate==0.31.0
bitsandbytes==0.43.1
ExifRead==3.0.0
imagesize==1.4.1
pillow==10.3.0
pyparsing==3.1.2
PySide6==6.7.2
transformers==4.41.2

# PyTorch
torch==2.2.2; platform_system != "Windows"
https://download.pytorch.org/whl/cu121/torch-2.2.2%2Bcu121-cp311-cp311-win_amd64.whl; platform_system == "Windows" and python_version == "3.11"
https://download.pytorch.org/whl/cu121/torch-2.2.2%2Bcu121-cp310-cp310-win_amd64.whl; platform_system == "Windows" and python_version == "3.10"

# FlashAttention (Florence-2)
flash-attn==2.5.9.post1; platform_system == "Linux"
https://github.com/oobabooga/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu122torch2.2.2cxx11abiFALSE-cp311-cp311-win_amd64.whl; platform_system == "Windows" and python_version == "3.11"
https://github.com/oobabooga/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu122torch2.2.2cxx11abiFALSE-cp310-cp310-win_amd64.whl; platform_system == "Windows" and python_version == "3.10"

# CogAgent
timm==1.0.7

# CogVLM
einops==0.8.0
protobuf==5.27.1
sentencepiece==0.2.0
torchvision==0.17.2
xformers==0.0.25.post1

# InternLM-XComposer2
auto-gptq==0.7.1; platform_system == "Linux" or platform_system == "Windows"
numpy==1.26.4

# WD Tagger
huggingface-hub==0.23.4
onnxruntime==1.18.0