Spaces:

ahmedmbutt
/

sd-gal-gen

Sleeping

ahmedmbutt commited on May 2, 2024

Commit

3fe0b34

1 Parent(s): d56c8ea

Update AutoPipelineForText2Image model name in lifespan function

Files changed (1) hide show

main.py CHANGED Viewed

@@ -15,7 +15,7 @@ from diffusers import (
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    text2img = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo").to(
         "cpu"
     )

 @asynccontextmanager
 async def lifespan(app: FastAPI):
+    text2img = AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo").to(
         "cpu"
     )