Update main.py
Browse files
main.py
CHANGED
@@ -17,7 +17,7 @@ app.add_middleware( # add the middleware
|
|
17 |
|
18 |
model_id = "runwayml/stable-diffusion-v1-5"
|
19 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
20 |
-
pipe = pipe.to("
|
21 |
|
22 |
def dummy(images, **kwargs):
|
23 |
return images, False
|
|
|
17 |
|
18 |
model_id = "runwayml/stable-diffusion-v1-5"
|
19 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
20 |
+
pipe = pipe.to("cpu")
|
21 |
|
22 |
def dummy(images, **kwargs):
|
23 |
return images, False
|