torch>=1.8.0 | |
torchvision>=0.9.0 | |
pytest>=6.0.0 | |
numpy>=1.19.0 | |
torchsummary>=1.5.1 | |
tqdm | |
matplotlib>=3.0.0 | |
diffusers==0.16.1 | |
# diffusers==0.21.4 | |
ftfy | |
# transformers==4.35.0 | |
transformers==4.25.1 | |
accelerate | |
safetensors | |
Pillow | |
huggingface-hub==0.25.2 | |
gradio | |
# opencv-python | |
scipy # Added for LMSDiscreteScheduler |