GenAI-Arena / requirements.txt
vinesmsuic's picture
Trigger rebuild
d7c6322 verified
flask
flask_restful
flask_cors
faiss-cpu
fire
h5py
xformers
numpy>=1.23.5
pandas<2.0.0
peft>=0.12
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
torchmetrics>=0.6.0
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
plotly
git+https://github.com/TIGER-AI-Lab/ImagenHub.git#egg=imagen-hub
git+https://github.com/TIGER-AI-Lab/VideoGenHub.git@arena#egg=videogen-hub
fal_client
open_clip_torch~=2.24.0
decord
huggingface_hub
open-clip-torch-any-py3
modelscope
protobuf==3.20.*
rotary_embedding_torch
av
oss2
simplejson
sortedcontainers
natsort
colossalai
mmengine
pre-commit
pyav
pyarrow
tensorboard
timm
bitsandbytes
wandb
pandarallel
kaleido