K-Sort-Arena / requirements.txt
YangZhoumill's picture
pre-release
6a771c3
flask
flask_restful
flask_cors
faiss-cpu
fire
h5py
# xformers~=0.0.20
numpy>=1.23.5
pandas<2.0.0
peft
torch==2.2.0
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.28.2
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
-e git+https://github.com/TIGER-AI-Lab/ImagenHub.git#egg=imagen-hub
fal_client
-e git+https://github.com/TIGER-AI-Lab/VideoGenHub.git#egg=videogen-hub
open_clip_torch
decord
huggingface_hub
open-clip-torch-any-py3
modelscope
protobuf==3.20.*
rotary_embedding_torch
av
natsort
colossalai
mmengine
pre-commit
pyav
pyarrow
tensorboard
timm
pandarallel
wandb
trueskill
replicate