Aneeshmishra commited on
Commit
f24ee52
·
verified ·
1 Parent(s): d1f836f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -2
requirements.txt CHANGED
@@ -1,8 +1,12 @@
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
7
  sentencepiece # tokenizer dependency
8
- gradio>=4.28
 
 
 
 
 
 
 
1
  transformers>=4.46.0
2
  optimum>=1.23.0 # makes transformers’ quantizer happy
3
  auto-gptq>=0.7.0.post0 # wheels available for CUDA 11.8/12.x
4
  bitsandbytes>=0.43.0 # still useful for 4-bit mat-muls
5
  accelerate>=0.29.0
6
  sentencepiece # tokenizer dependency
7
+ gradio>=4.28
8
+ # requirements.txt – CPU-only
9
+ transformers>=4.46.0 # model & pipeline
10
+ torch>=2.2.0 # CPU wheel auto-selected
11
+ sentencepiece>=0.1.99 # DistilBART tokenizer
12
+ gradio>=4.28.0 # UI