IBLlama_v1 / requirements.txt
tezuesh's picture
Upload folder using huggingface_hub
c80e261 verified
raw
history blame contribute delete
656 Bytes
torch==2.3.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.18.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