6psgvna / requirements.txt
pruning's picture
Upload folder using huggingface_hub
9f5d577 verified
raw
history blame contribute delete
673 Bytes
torch==2.6.0
sentencepiece==0.2.0
tiktoken==0.4.0
torchtune @ git+https://github.com/pytorch/torchtune.git@8f59c2fecd722691271eecca630a526719a32f76#egg=torchtune
lm_eval==0.4
torchvision==0.21.0
diffusers==0.27.2
imagebind @ git+https://github.com/omegalabsinc/ImageBind.git@c3c3b2e1ce6fd850ff42ce0375823fe22880a7cc#egg=imagebind
llama3 @ git+https://github.com/meta-llama/llama3.git@af6eedf7042fb51d00b2b26d8ef1ceaab73e1670
pytorchvideo @ git+https://github.com/facebookresearch/pytorchvideo.git@28fe037d212663c6a24f373b94cc5d478c8c1a1d
wandb==0.17.1
numpy==1.26.4
huggingface-hub==0.24.0
omegaconf==2.3.0
uvicorn==0.25.0
fastapi==0.104.1
pydantic==2.5.2
torchaudio==2.6.0