shivam9980 commited on
Commit
1050c45
1 Parent(s): cf6bb41

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -13
requirements.txt CHANGED
@@ -1,13 +1,5 @@
1
- unsloth[cu121] @ git+https://github.com/unslothai/unsloth.git
2
- streamlit
3
- trl==0.5.1 --no-deps
4
- peft==0.3.0.dev0 --no-deps
5
- accelerate==0.16.0 --no-deps
6
- bitsandbytes==0.37.2 --no-deps
7
- pytorch-cuda==<12.1/11.8>
8
- pytorch
9
- cudatoolkit
10
- xformers
11
- -c pytorch
12
- -c nvidia
13
- -c xformers
 
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/ --no-deps 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/ --no-deps torch torch-cuda torchvision torchaudio
4
+ -f https://huggingface.co/transformers/ --no-index --extra-index-url https://pypi.org/simple/ --no-deps transformers
5
+ -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