Upload .space\config.json with huggingface_hub
Browse files- .space//config.json +31 -0
.space//config.json
ADDED
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"runtime": {
|
3 |
+
"target": "zero",
|
4 |
+
"resources": {
|
5 |
+
"gpu": {
|
6 |
+
"vendor": "nvidia",
|
7 |
+
"model": "h200",
|
8 |
+
"count": 1
|
9 |
+
}
|
10 |
+
},
|
11 |
+
"env": {
|
12 |
+
"PYTHONPATH": "/home/user/app",
|
13 |
+
"CUDA_VISIBLE_DEVICES": "0",
|
14 |
+
"PYTORCH_CUDA_ALLOC_CONF": "max_split_size_mb:512",
|
15 |
+
"SPACES_ZERO_CUDA_INIT": "1",
|
16 |
+
"SPACES_ZERO_CUDA_VISIBLE_DEVICES": "0",
|
17 |
+
"SPACES_ZERO_CUDA_MEMORY_FRACTION": "0.95",
|
18 |
+
"SPACES_ZERO_ENABLE_QUEUE": "1",
|
19 |
+
"SPACES_ZERO_MAX_JOBS": "1"
|
20 |
+
}
|
21 |
+
},
|
22 |
+
"build": {
|
23 |
+
"python": "3.11",
|
24 |
+
"cuda": "12.1",
|
25 |
+
"env": {
|
26 |
+
"TORCH_CUDA_ARCH_LIST": "8.0;8.6;8.9;9.0",
|
27 |
+
"FORCE_CUDA": "1",
|
28 |
+
"TORCH_USE_CUDA_DSA": "1"
|
29 |
+
}
|
30 |
+
}
|
31 |
+
}
|