saliencies / requirements.txt
de-Rodrigo's picture
Modify to use Hugging Face ZeRO GPU
1dd7eb5
raw
history blame
153 Bytes
gradio>=4.0.0
transformers
huggingface_hub
torch>=2.0.0,!=2.3.0,<2.5.0
numpy
Pillow
datasets
deepspeed
huggingface-hub[cli,torch]
huggingface_hub[spaces]