ljsabc commited on
Commit
ee48564
β€’
1 Parent(s): fd095be

Update reqs.

Browse files
Files changed (1) hide show
  1. requirements.txt +11 -4
requirements.txt CHANGED
@@ -1,13 +1,20 @@
 
 
 
 
1
  # chatglm
2
  protobuf>=3.19.5,<3.20.1
3
  transformers==4.27.1
4
  icetk
5
  cpm_kernels==1.0.11
6
- torch>=1.13.1
7
-
8
 
 
9
  datasets==2.10.1
10
  git+https://github.com/huggingface/peft.git # ζœ€ζ–°η‰ˆζœ¬ >=0.3.0.dev0
11
 
12
-
13
-
 
 
 
 
 
1
+ # int8
2
+ bitsandbytes==0.37.1
3
+ accelerate==0.17.1
4
+
5
  # chatglm
6
  protobuf>=3.19.5,<3.20.1
7
  transformers==4.27.1
8
  icetk
9
  cpm_kernels==1.0.11
 
 
10
 
11
+ #
12
  datasets==2.10.1
13
  git+https://github.com/huggingface/peft.git # ζœ€ζ–°η‰ˆζœ¬ >=0.3.0.dev0
14
 
15
+ -f https://download.pytorch.org/whl/cpu
16
+ torch
17
+ -f https://download.pytorch.org/whl/cpu
18
+ torchvision
19
+ -f https://download.pytorch.org/whl/cpu
20
+ torchaudio