shivam9980 commited on
Commit
f9bcc94
1 Parent(s): 8623e44

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -5
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
- unsloth[cu121] @ git+https://github.com/unslothai/unsloth.git@streamlittrl==0.5.1
2
- -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
3
- -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
4
- -f https://huggingface.co/transformers/ --no-index --extra-index-url https://pypi.org/simple/ transformers
5
- -f https://developer.nvidia.com/compute/cuda/repos/pypi/ --index-url https://pypi.org/simple/ --extra-index-url https://pypi.org/simple/ nvidia
 
 
1
+ unsloth[cu121] @ git+https://github.com/unslothai/unsloth.git
2
+ conda install pytorch-cuda=<12.1/11.8> pytorch cudatoolkit xformers -c pytorch -c nvidia -c xformers
3
+ trl
4
+ peft
5
+ accelerate
6
+ bitsandbytes