Spaces:
Sleeping
Sleeping
Factor Studios
commited on
Update test_ai_integration_http.py
Browse files
test_ai_integration_http.py
CHANGED
|
@@ -248,7 +248,13 @@ def test_ai_integration_http():
|
|
| 248 |
"model_type": "placeholder",
|
| 249 |
"parameters": 1000000, # Small placeholder
|
| 250 |
"architecture": "test",
|
| 251 |
-
"loaded_at": time.time()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 252 |
}
|
| 253 |
|
| 254 |
success = ai_accelerator_for_loading.load_model(
|
|
|
|
| 248 |
"model_type": "placeholder",
|
| 249 |
"parameters": 1000000, # Small placeholder
|
| 250 |
"architecture": "test",
|
| 251 |
+
"loaded_at": time.time(),
|
| 252 |
+
# Required GPU architecture configuration
|
| 253 |
+
"num_sms": 108, # Similar to A100
|
| 254 |
+
"tensor_cores_per_sm": 4,
|
| 255 |
+
"cuda_cores_per_sm": 64,
|
| 256 |
+
"compute_capability": "8.0",
|
| 257 |
+
"vram_allocation": "dynamic"
|
| 258 |
}
|
| 259 |
|
| 260 |
success = ai_accelerator_for_loading.load_model(
|