shivam9980 commited on
Commit
55fa764
1 Parent(s): ce44292

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +12 -2
requirements.txt CHANGED
@@ -1,3 +1,13 @@
1
- transformers @ git+https://github.com/huggingface/transformers.git
2
  streamlit
3
- peft
 
 
 
 
 
 
 
 
 
 
 
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