TF-ICON-unofficial / requirements.txt
sky24h's picture
init commit
910b9ab
raw
history blame contribute delete
470 Bytes
torch==2.1.1
torchvision==0.16.1
xformers==0.0.23
gradio==4.44.1
numpy==1.23.1
albumentations==1.3.0
opencv-python==4.6.0.66
imageio==2.9.0
imageio-ffmpeg==0.4.2
pytorch-lightning==1.4.2
omegaconf==2.1.1
test-tube>=0.7.5
streamlit==1.12.1
einops==0.3.0
transformers==4.19.2
webdataset==0.2.5
kornia==0.6
open_clip_torch==2.0.2
invisible-watermark>=0.1.5
streamlit-drawable-canvas==0.8.0
torchmetrics==0.6.0
diffusers==0.12.1
ipykernel
matplotlib
gradio_image_annotation