VanguardAI commited on
Commit
a6c240f
·
verified ·
1 Parent(s): e238b8b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -9
requirements.txt CHANGED
@@ -1,11 +1,10 @@
1
- unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git
2
- packaging
3
- ninja
4
- einops
5
- flash-attn
6
- xformers
7
- trl
8
- peft
9
  accelerate
10
  bitsandbytes
11
- torch==2.2.1
 
 
1
+ pip==24.1.2
2
+ --find-links https://download.pytorch.org/whl/cu121
3
+ torch==2.2.0
4
+ triton
5
+ git+https://github.com/unslothai/unsloth.git#egg=unsloth[cu121-ampere-torch220]
6
+ transformers
 
 
7
  accelerate
8
  bitsandbytes
9
+ einops
10
+ xformers