GenAI-Arena / requirements.txt
yuanshengni's picture
Init
a635e94
raw
history blame
873 Bytes
flask
flask_restful
flask_cors
faiss-cpu
fire
h5py
xformers~=0.0.20
numpy>=1.24.0
pandas<2.0.0
peft
torch
torchvision
torchaudio
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
gradio==4.21.0
pillow~=9.5.0
einops>=0.4.1
torch-fidelity>=0.3.0
setuptools>=59.5.0
transformers==4.37.2
torchmetrics>=0.6.0
lpips
dreamsim
image-reward
kornia>=0.6
diffusers>=0.18.0
accelerate>=0.20.3
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
-e git+https://github.com/TIGER-AI-Lab/ImagenHub.git#egg=imagen-hub
plotly