shivam9980-mistral-7b-news / requirements.txt
shivam9980's picture
Update requirements.txt
1050c45 verified
raw
history blame
706 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/ --no-deps accelerate==0.16.0
-f https://dl.fbaipublicfiles.com/pytorch/channel/stable/cu121/index.html --no-index --extra-index-url https://pypi.org/simple/ --no-deps torch torch-cuda torchvision torchaudio
-f https://huggingface.co/transformers/ --no-index --extra-index-url https://pypi.org/simple/ --no-deps transformers
-f https://developer.nvidia.com/compute/cuda/repos/pypi/ --index-url https://pypi.org/simple/ --extra-index-url https://pypi.org/simple/ --no-deps nvidia