tvar-demo-test-2 / requirements.txt
michellemoorre's picture
Add reqs and preload
fb29620
raw
history blame
394 Bytes
apex==0.9.10dev
decord==0.6.0
einops==0.8.0
flash_attn==2.6.3
huggingface_hub==0.26.1
image_reward==1.5
mpi4py==4.0.1
nirvana_dl==66.0.3
numpy==2.1.2
omegaconf==2.3.0
pandas==2.2.3
Pillow==11.0.0
pytz==2024.2
PyYAML==6.0.2
PyYAML==6.0.2
scipy==1.14.1
tensorflow_io==0.37.1
torch==2.5.0
torchvision==0.20.0
tqdm==4.66.5
transformers==4.45.2
typed_argument_parser==1.10.1
wandb==0.18.5
yt==4.3.1