Aneeshmishra commited on
Commit
05b7b67
·
verified ·
1 Parent(s): 7e7fb74

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -4
requirements.txt CHANGED
@@ -1,4 +1,6 @@
1
- transformers>=4.41
2
- accelerate
3
- bitsandbytes
4
- einops
 
 
 
1
+ torch>=2.1
2
+ transformers>=4.46.0
3
+ optimum>=1.23.0 # makes transformers’ quantizer happy
4
+ auto-gptq>=0.7.0.post0 # wheels available for CUDA 11.8/12.x
5
+ bitsandbytes>=0.43.0 # still useful for 4-bit mat-muls
6
+ accelerate>=0.29.0