Finetuned_Diffusion_Max / requirements.txt
ReiPlush64's picture
Update requirements.txt
36c3ea3 verified
raw
history blame
566 Bytes
--extra-index-url https://download.pytorch.org/whl/cu117
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
torchvision==0.14.1+cu117
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.15