Suporte commited on
Commit
dfafb32
·
1 Parent(s): 756cb50

Use CPU-only PyTorch wheel to keep CPU Basic build lean

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -1,7 +1,9 @@
 
 
1
  gradio==5.29.1
2
- torch>=2.2.0
3
  diffusers>=0.35.0
4
- transformers>=4.46.0
5
  accelerate>=1.0.0
6
  safetensors>=0.4.5
7
  Pillow>=10.0.0
 
1
+ --extra-index-url https://download.pytorch.org/whl/cpu
2
+
3
  gradio==5.29.1
4
+ torch==2.5.1+cpu
5
  diffusers>=0.35.0
6
+ transformers>=4.46.0,<5
7
  accelerate>=1.0.0
8
  safetensors>=0.4.5
9
  Pillow>=10.0.0