ReVQ / requirements.txt
AndyRaoTHU's picture
debug
e7b8988
raw
history blame contribute delete
317 Bytes
torch>=2.3.0
torchvision
torchaudio
numpy
matplotlib
scikit-learn
opencv-python
einops>=0.7.0
omegaconf>=2.3.0
lightning>=2.1.3
transformers>=4.36.2
pudb
wandb
tensorboard
termcolor
lpips==0.1.4
ftfy
imageio
imageio-ffmpeg
pyiqa
clean-fid
albumentations
pytorch-fid
torchinfo
gradio==4.15.0
diffusers
pydantic==2.10.6