Werli commited on
Commit
4af4674
·
verified ·
1 Parent(s): 2eec19d

More speed

Browse files

Changes:
Added `--extra-index-url https://download.pytorch.org/whl/cu121` to accelerate packages installation.
Downgraded Torch to `2.5.0` to accelerate image tagging.

Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
- torch
 
2
  gradio==5.50.0
3
  safetensors
4
  sentencepiece
@@ -14,6 +15,5 @@ onnxruntime>=1.12.0
14
  matplotlib
15
  apscheduler
16
  spaces
17
- pandas
18
  huggingface-hub
19
  transformers
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu121
2
+ torch==2.5.0
3
  gradio==5.50.0
4
  safetensors
5
  sentencepiece
 
15
  matplotlib
16
  apscheduler
17
  spaces
 
18
  huggingface-hub
19
  transformers