[SHUTDOWN{}

The-Swarm-Corporation · Mar 20, 2024 · e134ab7 · e134ab7
1 parent fa5bec5
commit e134ab7
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms-cloud"
-version = "0.1.3"
+version = "0.1.6"
 description = "Swarms Cloud - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <[email protected]>"]
@@ -30,7 +30,10 @@ torch = "*"
 einops = "*"
 pydantic = "*"
 stripe = "*"
+transformers = "*"
 sse-starlette = "*" 
+uvicorn = "*"
+shortuuid = "*"
 
 [tool.poetry.group.lint.dependencies]
 ruff = ">=0.1.6,<0.4.0"

diff --git a/servers/cogvlm/cogvlm.py b/servers/cogvlm/cogvlm.py
@@ -78,7 +78,7 @@ async def lifespan(app: FastAPI):
 
 
 # On startup
-@app.on_event("startup")
+@app.on_startup()
 async def load_model():
     global model, tokenizer, torch_type, QUANT_ENABLED
 
@@ -431,7 +431,7 @@ def generate_stream_cogvlm(
     yield ret
 
 
-@app.on_event("shutdown")
+@app.on_shutdown()
 async def shutdown_event():
     print("Application shutdown, cleaning up artifacts")
     try:

diff --git a/servers/cogvlm/cog_vlm.py → servers/cogvlm/example_cogvlm_api_request.py b/servers/cogvlm/cog_vlm.py → servers/cogvlm/example_cogvlm_api_request.py