Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ def load_pipeline():
|
|
12 |
)
|
13 |
pipe.to("cuda" if torch.cuda.is_available() else "cpu")
|
14 |
|
15 |
-
pipe.scheduler = DPMSolverMultistepScheduler.from_config(
|
16 |
|
17 |
pipe.enable_attention_slicing()
|
18 |
|
|
|
12 |
)
|
13 |
pipe.to("cuda" if torch.cuda.is_available() else "cpu")
|
14 |
|
15 |
+
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
16 |
|
17 |
pipe.enable_attention_slicing()
|
18 |
|