shivam9980-mistral-7b-news / requirements.txt
shivam9980's picture
Update requirements.txt
8623e44 verified
raw
history blame
668 Bytes
unsloth[cu121] @ git+https://github.com/unslothai/unsloth.git@streamlittrl==0.5.1
-f https://accelerate.huggingface.co/channels/huggingface-hub/packages/accelerate/index.html --no-index --extra-index-url https://pypi.org/simple/ accelerate==0.16.0
-f https://dl.fbaipublicfiles.com/pytorch/channel/stable/cu121/index.html --no-index --extra-index-url https://pypi.org/simple/ torch torch-cuda torchvision torchaudio
-f https://huggingface.co/transformers/ --no-index --extra-index-url https://pypi.org/simple/ transformers
-f https://developer.nvidia.com/compute/cuda/repos/pypi/ --index-url https://pypi.org/simple/ --extra-index-url https://pypi.org/simple/ nvidia