waifu-diffusion/requirements.txt

numpy==1.21.6
albumentations==0.4.3
opencv-python
pudb==2019.2
imageio==2.9.0
imageio-ffmpeg==0.4.2
pytorch-lightning==1.7.7
omegaconf==2.1.1
test-tube>=0.7.5
streamlit>=0.73.1
einops==0.3.0
torch-fidelity==0.3.0
transformers==4.19.2
diffusers==0.7.1
torchmetrics==0.7.0
kornia==0.6
gradio
git+https://github.com/illeatmyhat/taming-transformers.git@master#egg=taming-transformers
git+https://github.com/openai/CLIP.git@main#egg=clip
git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion
webdataset
wandb
fairscale
pynvml==11.4.1
make the execution work 2022-09-17 00:55:56 -06:00			`numpy==1.21.6`
add requirements 2022-08-25 16:06:34 -06:00			`albumentations==0.4.3`
make the execution work 2022-09-17 00:55:56 -06:00			`opencv-python`
add requirements 2022-08-25 16:06:34 -06:00			`pudb==2019.2`
			`imageio==2.9.0`
			`imageio-ffmpeg==0.4.2`
add fp16 training 2022-10-15 11:57:00 -06:00			`pytorch-lightning==1.7.7`
add requirements 2022-08-25 16:06:34 -06:00			`omegaconf==2.1.1`
			`test-tube>=0.7.5`
			`streamlit>=0.73.1`
			`einops==0.3.0`
			`torch-fidelity==0.3.0`
			`transformers==4.19.2`
Add missing reqs / fix versions 2022-11-04 18:31:08 -06:00			`diffusers==0.7.1`
			`torchmetrics==0.7.0`
add requirements 2022-08-25 16:06:34 -06:00			`kornia==0.6`
			`gradio`
Update requirements.txt 2022-09-15 11:44:08 -06:00			`git+https://github.com/illeatmyhat/taming-transformers.git@master#egg=taming-transformers`
add requirements 2022-08-25 16:06:34 -06:00			`git+https://github.com/openai/CLIP.git@main#egg=clip`
Update requirements.txt Added missing "#egg" requirement name for K diffusion. Changed K diffusion repo, previous repo removed "setup.py" and would error out. New repository added still works. 2022-08-27 03:53:34 -06:00			`git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion`
add fp16 training 2022-10-15 11:57:00 -06:00			`webdataset`
			`wandb`
Add missing reqs / fix versions 2022-11-04 18:31:08 -06:00			`fairscale`
			`pynvml==11.4.1`