Spaces:
Sleeping
Sleeping
Factor Studios
commited on
Update virtual_gpu_server_http.py
Browse files
virtual_gpu_server_http.py
CHANGED
|
@@ -635,7 +635,7 @@ def sanitize_filename(name: str) -> str:
|
|
| 635 |
"""
|
| 636 |
return name.replace('/', '__')
|
| 637 |
|
| 638 |
-
@app.post("/api/v1/models/{model_name}/load")
|
| 639 |
async def load_model(
|
| 640 |
model_name: str,
|
| 641 |
request: ModelLoadRequest,
|
|
@@ -684,7 +684,7 @@ async def load_model(
|
|
| 684 |
detail=f"Model load operation failed: {str(e)}"
|
| 685 |
)
|
| 686 |
|
| 687 |
-
@app.post("/api/v1/models/{model_name}/inference")
|
| 688 |
async def run_inference(
|
| 689 |
model_name: str,
|
| 690 |
request: ModelInferenceRequest,
|
|
@@ -734,7 +734,7 @@ async def run_inference(
|
|
| 734 |
detail=f"Inference operation failed: {str(e)}"
|
| 735 |
)
|
| 736 |
|
| 737 |
-
@app.get("/api/v1/models/{model_name}/status")
|
| 738 |
async def get_model_status(
|
| 739 |
model_name: str,
|
| 740 |
session: Dict[str, Any] = Depends(get_current_session)
|
|
|
|
| 635 |
"""
|
| 636 |
return name.replace('/', '__')
|
| 637 |
|
| 638 |
+
@app.post("/api/v1/models/{model_name:path}/load")
|
| 639 |
async def load_model(
|
| 640 |
model_name: str,
|
| 641 |
request: ModelLoadRequest,
|
|
|
|
| 684 |
detail=f"Model load operation failed: {str(e)}"
|
| 685 |
)
|
| 686 |
|
| 687 |
+
@app.post("/api/v1/models/{model_name:path}/inference")
|
| 688 |
async def run_inference(
|
| 689 |
model_name: str,
|
| 690 |
request: ModelInferenceRequest,
|
|
|
|
| 734 |
detail=f"Inference operation failed: {str(e)}"
|
| 735 |
)
|
| 736 |
|
| 737 |
+
@app.get("/api/v1/models/{model_name:path}/status")
|
| 738 |
async def get_model_status(
|
| 739 |
model_name: str,
|
| 740 |
session: Dict[str, Any] = Depends(get_current_session)
|