Finetuned_Diffusion_Max / requirements.txt
ReiPlush64's picture
Update requirements.txt
c4e2e10 verified
--extra-index-url https://download.pytorch.org/whl/cu113
addict==2.4.0
albumentations==1.3.0
einops==0.6.0
gradio==3.21
huggingface-hub
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.1
timm
torchvision==0.14.1
fastai
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
xformers==0.0.16