Diffree / requirements.txt
cocktailpeanut's picture
update
41a3c74
raw
history blame contribute delete
483 Bytes
#--extra-index-url https://download.pytorch.org/whl/cu117
numpy==1.24.4
#torch==2.0.0
#torchvision==0.15.1
torchmetrics==0.6.0
pytorch-lightning==1.4.2
transformers==4.19.2
tqdm==4.66.2
gradio==3.50.2
openai==1.12.0
opencv-python
einops==0.3.0
omegaconf==2.1.1
-e git+https://github.com/crowsonkb/[email protected]#egg=k-diffusion
-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
imageio==2.9.0
imageio-ffmpeg==0.4.2
#spaces
tokenizers