IdlecloudX commited on
Commit
2109375
·
verified ·
1 Parent(s): 929a9dd

Use matching CPU PyTorch builds on the CPU-only Space

Browse files
Files changed (1) hide show
  1. requirements.txt +16 -14
requirements.txt CHANGED
@@ -1,14 +1,16 @@
1
- gradio==5.29.0
2
- fastapi>=0.115.2,<1
3
- huggingface_hub>=0.34,<1.0
4
- numpy
5
- pandas
6
- Pillow>=10
7
- onnxruntime
8
- torch>=2.6,<3
9
- torchvision>=0.21,<1
10
- timm==1.0.27
11
- transformers==4.57.6
12
- requests
13
- httpx==0.28.1
14
- pydantic>=2,<2.12
 
 
 
1
+ gradio==5.29.0
2
+ fastapi>=0.115.2,<1
3
+ huggingface_hub>=0.34,<1.0
4
+ numpy
5
+ pandas
6
+ Pillow>=10
7
+ onnxruntime
8
+ # 此空间使用 CPU;固定同版 CPU 构建,避免下载无用的 CUDA 运行库。
9
+ --extra-index-url https://download.pytorch.org/whl/cpu
10
+ torch==2.14.1+cpu
11
+ torchvision==0.29.1+cpu
12
+ timm==1.0.27
13
+ transformers==4.57.6
14
+ requests
15
+ httpx==0.28.1
16
+ pydantic>=2,<2.12