--extra-index-url https://download.pytorch.org/whl/cu124
torch==2.1.0
torchvision==0.16.0
numpy<2
diffusers==0.32.1
accelerate==1.1.1
transformers==4.46.2
huggingface-hub==0.25.0 
tensorboard
gradio
torchdiffeq
click
torchvision
opencv-python
scikit-image
numba
scipy
tqdm
einops
sentencepiece
hf_xet