Finetuned_Diffusion_Max / requirements.txt
SUPERSHANKY's picture
Update requirements.txt
e5bafc5
raw
history blame
695 Bytes
--extra-index-url https://download.pytorch.org/whl/cu118
addict==2.4.0
albumentations==1.3.0
einops==0.6.0
gradio==3.21
huggingface-hub==0.13.2
imageio==2.26.0
imageio-ffmpeg==0.4.8
kornia
omegaconf==2.3.0
open-clip-torch
opencv-contrib-python==4.7.0.72
opencv-python-headless==4.7.0.72
prettytable==3.6.0
pytorch-lightning
safetensors==0.3.0
timm
https://download.pytorch.org/whl/cu118/torch-2.0.0%2Bcu118-cp39-cp39-linux_x86_64.whl
https://download.pytorch.org/whl/cu118/torchvision-0.15.1%2Bcu118-cp39-cp39-linux_x86_64.whl
transformers==4.27.1
datasets==2.9.0
accelerate
evaluate==0.4.0
yapf==0.32.0
Triton==2.0.0
#diffusers
git+https://github.com/huggingface/diffusers.git
scipy
ftfy
psutil