visual-arena / requirements.txt
tianleliphoebe's picture
Upload folder using huggingface_hub
ec0c335 verified
raw
history blame
844 Bytes
transformers
torch
gradio~=3.50.2
xformers~=0.0.20
uvicorn
accelerate
diffusers~=0.20.2
pillow
-e git+https://github.com/TIGER-AI-Lab/ImagenHub.git#egg=imagen-hub
torchvision
peft
torchaudio
opencv-python
jupyterlab>=4.0.2
notebook>=6.5.4
albumentations>=1.1.0
opencv-python>=4.2.0
pudb~=2019.2
imageio>=2.14.1
imageio-ffmpeg>=0.4.7
pytorch-lightning>=1.5.9
omegaconf~=2.1.1
einops>=0.4.1
torch-fidelity>=0.3.0
setuptools>=59.5.0
torchmetrics>=0.6.0
lpips
dreamsim
image-reward
kornia>=0.6
safetensors
datasets
tqdm>=4.64.1
matplotlib>=3.7.1
taming-transformers-rom1504~=0.0.6
madgrad>=1.1
-e git+https://github.com/openai/CLIP.git@main#egg=clip
dominate>=2.8.0
-e git+https://github.com/CompVis/latent-diffusion.git#egg=latent-diffusion #ldm
openai
nltk~=3.8.1
krippendorff
statsmodels
flask
flask_restful
flask_cors
faiss-cpu
fire
h5py
plotly