# --extra-index-url https://download.pytorch.org/whl/cu124
# torch==2.6.0
# torchvision==0.21.0
# torchaudio==2.6.0

gradio>=4.0
accelerate==0.30.1
deepspeed==0.14.4
einops==0.8.0
transformers==4.43.3
huggingface-hub==0.24.5
optimum-quanto
datasets
omegaconf
diffusers
sentencepiece
opencv-python
matplotlib
onnxruntime
timm
wandb

setuptools
wheel