NLPAlhuzali commited on
Commit
ace995d
Β·
verified Β·
1 Parent(s): f291881

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -21
requirements.txt CHANGED
@@ -1,29 +1,11 @@
1
- # core
2
  torch>=2.1.0
3
  transformers>=4.40.0
4
  accelerate>=0.27.2
5
-
6
- # model-speed / memory
 
7
  bitsandbytes>=0.43.0
8
  xformers>=0.0.26
9
  einops>=0.7.0
10
-
11
- # interface
12
- gradio>=4.31.0
13
-
14
- # notebook extras (present in your original code)
15
  pandas>=2.2.0
16
  numpy>=1.26.0
17
-
18
- # tokenizer helpers – add these two lines
19
- sentencepiece>=0.1.99
20
- tiktoken>=0.6.0
21
-
22
-
23
-
24
- # Yasser's reqs
25
- torch # HF will pick a CUDA / CPU build automatically
26
- transformers>=4.39
27
- accelerate # lets Transformers shard the model
28
- sentencepiece # <-- needed for the tokenizer
29
- gradio==4.18
 
 
1
  torch>=2.1.0
2
  transformers>=4.40.0
3
  accelerate>=0.27.2
4
+ sentencepiece>=0.1.99
5
+ tiktoken>=0.6.0
6
+ gradio>=4.31.0
7
  bitsandbytes>=0.43.0
8
  xformers>=0.0.26
9
  einops>=0.7.0
 
 
 
 
 
10
  pandas>=2.2.0
11
  numpy>=1.26.0