Update app.py
Browse files
app.py
CHANGED
|
@@ -134,6 +134,8 @@ def infer(
|
|
| 134 |
|
| 135 |
pipe.to(device)
|
| 136 |
|
|
|
|
|
|
|
| 137 |
image = pipe(**pipe_params).images[0]
|
| 138 |
|
| 139 |
return image, seed
|
|
|
|
| 134 |
|
| 135 |
pipe.to(device)
|
| 136 |
|
| 137 |
+
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
| 138 |
+
|
| 139 |
image = pipe(**pipe_params).images[0]
|
| 140 |
|
| 141 |
return image, seed
|