mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-23 04:12:13 +00:00
bug fix
This commit is contained in:
parent
bde8503620
commit
c9b440e381
@ -95,7 +95,7 @@ async def api_generate_stream(request: Request):
|
||||
@app.post("/generate")
|
||||
def generate(prompt_request: Request):
|
||||
|
||||
params = request.json()
|
||||
params = prompt_request.json()
|
||||
print("Receive prompt: ", params["prompt"])
|
||||
output = generate_output(model, tokenizer, params, DEVICE)
|
||||
print("Output: ", output)
|
||||
|
Loading…
Reference in New Issue
Block a user