visconet / requirements.txt
soonyau's picture
Update requirements.txt
d76475d verified
raw
history blame contribute delete
572 Bytes
pip==20.3
torch==1.12.1
torchvision==0.13.1
numpy==1.23.1
git+https://github.com/openai/CLIP.git
gradio==3.39.0
albumentations==1.3.0
imageio==2.9.0
imageio-ffmpeg==0.4.2
pytorch-lightning==1.5.0
omegaconf==2.1.1
test-tube>=0.7.5
einops==0.3.0
transformers==4.27.4
webdataset==0.2.5
kornia==0.6
open_clip_torch==2.0.2
invisible-watermark>=0.1.5
torchmetrics==0.6.0
timm==0.6.12
addict==2.4.0
yapf==0.32.0
prettytable==3.6.0
safetensors==0.2.7
triton
basicsr==1.4.2
xformers==0.0.13
#deepface
#opencv-contrib-python
--extra-index-url https://download.pytorch.org/whl/cu113