Skip to content

GitLab

"git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "2e1d2d7e66c33fdd2b58aaf03a9893dbe593a3a3"
Commit 23c64538 authored by Jeffrey Morgan's avatar Jeffrey Morgan
Browse files

fix model lookup directory for other routes

parent 2db9c1bb
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
proto.py
View file @ 23c64538
...@@ -18,11 +18,11 @@ def load(): ...@@ -18,11 +18,11 @@ def load():
if not model: if not model:
return Response("Model is required", status=400) return Response("Model is required", status=400)
if not os.path.exists(f"../models/{model}.bin"): if not os.path.exists(f"./models/{model}.bin"):
return {"error": "The model does not exist."}, 400 return {"error": "The model does not exist."}, 400
if model not in llms: if model not in llms:
llms[model] = Llama(model_path=f"../models/{model}.bin") llms[model] = Llama(model_path=f"./models/{model}.bin")
return Response(status=204) return Response(status=204)
...@@ -34,7 +34,7 @@ def unload(): ...@@ -34,7 +34,7 @@ def unload():
if not model: if not model:
return Response("Model is required", status=400) return Response("Model is required", status=400)
if not os.path.exists(f"../models/{model}.bin"): if not os.path.exists(f"./models/{model}.bin"):
return {"error": "The model does not exist."}, 400 return {"error": "The model does not exist."}, 400
llms.pop(model, None) llms.pop(model, None)
...@@ -57,7 +57,7 @@ def generate(): ...@@ -57,7 +57,7 @@ def generate():
if model not in llms: if model not in llms:
# auto load # auto load
llms[model] = Llama(model_path=f"../models/{model}.bin") llms[model] = Llama(model_path=f"./models/{model}.bin")
def stream_response(): def stream_response():
stream = llms[model]( stream = llms[model](
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment