test / requirements.txt
silentchen's picture
Upload requirements.txt
6ae5687
raw
history blame contribute delete
362 Bytes
torch==1.13.1
torchvision==0.14.1
omegaconf==2.1.1
albumentations==1.3.0
opencv-python
imageio==2.9.0
imageio-ffmpeg==0.4.2
pytorch-lightning==1.4.2
test-tube>=0.7.5
streamlit==1.17.0
einops==0.3.0
protobuf~=3.20.1
torchmetrics==0.6.0
transformers==4.24.0
kornia==0.6.0
gradio==3.19.1
diffusers==0.7.2
accelerate==0.13.2
git+https://github.com/openai/CLIP.git