fix api

Former-commit-id: 4c3e8be325045e432b31c519132123c7b0689262
2023-07-20 22:14:54 +08:00
parent 5dcd95645f
commit 2006a96570
2 changed files with 5 additions and 2 deletions
--- a/src/llmtuner/api/app.py
+++ b/src/llmtuner/api/app.py
@@ -30,7 +30,7 @@ async def lifespan(app: FastAPI): # collects GPU memory
    torch_gc()


-def create_app(chat_model: ChatModel):
+def create_app(chat_model: ChatModel) -> FastAPI:
    app = FastAPI(lifespan=lifespan)

    app.add_middleware(