requirements.txt

accelerate==1.6.0
diffusers==0.33.1
einops==0.8.1
omegaconf==2.3.0
opencv-python==4.11.0.86
peft==0.15.2
qwen-vl-utils==0.0.11
scikit-learn==1.6.1
kornia==0.8.0
webdataset==0.2.111
segment-anything==1.0
tensorboard==2.19.0
tensorboardx==2.6.2.2
transformers==4.56.0
gradio==5.43.1
tqdm==4.67.1
torch==2.8.0
    # via
    #   -r requirements.in
    #   accelerate
    #   deepspeed
    #   lightning
    #   peft
    #   pytorch-lightning
    #   torchvision
sentencepiece==0.2.0
huggingface-hub==0.34.4
safetensors==0.5.3
open_clip_torch==3.1.0
timm==1.0.19