qqqcat commited on
Commit
ec5f6a4
·
verified ·
1 Parent(s): f4610f6

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +9 -4
requirements.txt CHANGED
@@ -1,5 +1,10 @@
1
- torch==2.1.0
2
- transformers>=4.37.0
3
- accelerate>=0.25.0
4
- gradio>=4.0.0
 
 
 
5
  sentencepiece
 
 
 
1
+ # CPU build
2
+ numpy==1.25.4
3
+ torch==2.1.0+cpu
4
+ -f https://download.pytorch.org/whl/cpu/torch_stable.html
5
+ transformers==4.37.2
6
+ accelerate==0.25.1
7
+ safetensors
8
  sentencepiece
9
+ gradio>=4.0
10
+ huggingface-hub