Factor Studios commited on
Commit
6b96572
·
verified ·
1 Parent(s): d276130

Update test_ai_integration_http.py

Browse files
Files changed (1) hide show
  1. test_ai_integration_http.py +7 -1
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(