Factor Studios commited on
Commit
20c4c8d
·
verified ·
1 Parent(s): f3885ff

Update virtual_gpu_server_http.py

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