ByteMorpher-Demo / requirements.txt
aducsdr's picture
Update requirements.txt
e74e2fb verified
raw
history blame contribute delete
348 Bytes
# --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