RishiRP commited on
Commit
e72eb67
·
verified ·
1 Parent(s): 7a1eb70

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,11 +1,11 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu118
2
 
3
- # Core compute (CUDA 11.8)
4
  torch==2.2.2+cu118
5
 
6
  # NLP stack
7
  transformers==4.43.3
8
- tokenizers==0.19.1 # compatible with 4.43.3; fixes tokenizer.json parse for Apertus
9
  accelerate==0.33.0
10
  safetensors==0.4.4
11
  bitsandbytes==0.43.1
 
1
  --extra-index-url https://download.pytorch.org/whl/cu118
2
 
3
+ # Core compute (CUDA 11.8, good for Tesla T4)
4
  torch==2.2.2+cu118
5
 
6
  # NLP stack
7
  transformers==4.43.3
8
+ tokenizers==0.19.1 # compatible with transformers 4.43.x; avoids tokenizer.json issue
9
  accelerate==0.33.0
10
  safetensors==0.4.4
11
  bitsandbytes==0.43.1