BMike10 commited on
Commit
288ddad
1 Parent(s): 16fd01b
Files changed (1) hide show
  1. requirements.txt +4 -16
requirements.txt CHANGED
@@ -1,16 +1,4 @@
1
- accelerate==0.28.0
2
- torch
3
- safetensors
4
- bitsandbytes==0.43.0
5
- scipy
6
- peft
7
- transformers
8
- tqdm
9
- packaging
10
- pytest
11
- numpy
12
- pyyaml
13
- datasets
14
- psutil
15
- setuptools
16
- bitsandbytes-cuda102
 
1
+ torch==2.0.1
2
+ transformers==4.34.0
3
+ bitsandbytes==0.41.1
4
+ accelerate==0.23.0