Finetuned_Diffusion_Max / requirements.txt
SUPERSHANKY's picture
Update requirements.txt
3fcf59e
raw
history blame
No virus
636 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-1.13.1%2Bcu118-cp39-cp39-linux_x86_64.whl
torchvision==0.14.1+cu118
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