Skip to content

Commit

Permalink
[REVERT]
Browse files Browse the repository at this point in the history
  • Loading branch information
Kye committed Mar 20, 2024
1 parent e134ab7 commit 8c264c4
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions servers/cogvlm/cogvlm.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ async def lifespan(app: FastAPI):


# On startup
@app.on_startup()
@app.on_event("startup")
async def load_model():
global model, tokenizer, torch_type, QUANT_ENABLED

Expand Down Expand Up @@ -431,7 +431,7 @@ def generate_stream_cogvlm(
yield ret


@app.on_shutdown()
@app.on_event("shutdown")
async def shutdown_event():
print("Application shutdown, cleaning up artifacts")
try:
Expand Down

0 comments on commit 8c264c4

Please sign in to comment.