Spaces:
Running
Running
Commit
•
3fe0b34
1
Parent(s):
d56c8ea
Update AutoPipelineForText2Image model name in lifespan function
Browse files
main.py
CHANGED
@@ -15,7 +15,7 @@ from diffusers import (
|
|
15 |
|
16 |
@asynccontextmanager
|
17 |
async def lifespan(app: FastAPI):
|
18 |
-
text2img = AutoPipelineForText2Image.from_pretrained("stabilityai/
|
19 |
"cpu"
|
20 |
)
|
21 |
|
|
|
15 |
|
16 |
@asynccontextmanager
|
17 |
async def lifespan(app: FastAPI):
|
18 |
+
text2img = AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo").to(
|
19 |
"cpu"
|
20 |
)
|
21 |
|