LaVie / requirements.txt
Zhouyan248's picture
Update requirements.txt
a876b6c
accelerate==0.24.1
av==10.0.0
decord==0.6.0
diffusers[torch]==0.16.0
ffmpeg==1.4
imageio==2.31.1
imageio-ffmpeg==0.4.9
pandas==2.0.1
timm==0.6.13
tqdm==4.65.0
transformers==4.35.0
xformers==0.0.20
omegaconf==2.3.0
natsort==8.4.0
rotary_embedding_torch
gradio==4.3.0
black==23.7.0
chardet==5.1.0
clip @ git+https://github.com/openai/CLIP.git
einops>=0.6.1
fairscale>=0.4.13
fire>=0.5.0
fsspec>=2023.6.0
invisible-watermark>=0.2.0
kornia==0.6.9
matplotlib>=3.7.2
natsort>=8.4.0
ninja>=1.11.1
numpy>=1.24.4
open-clip-torch>=2.20.0
opencv-python==4.6.0.66
pillow>=9.5.0
pudb>=2022.1.3
pytorch-lightning==2.0.1
pyyaml>=6.0.1
scipy>=1.10.1
streamlit>=0.73.1
tensorboardx==2.6
tokenizers==0.14.1
torch>=2.0.1
torchaudio>=2.0.2
torchdata==0.6.1
torchmetrics>=1.0.1
torchvision>=0.15.2
triton==2.0.0
urllib3<1.27,>=1.25.4
wandb>=0.15.6
webdataset>=0.2.33
wheel>=0.41.0
fire
uuid