Docty commited on
Commit
8f4f59a
·
verified ·
1 Parent(s): 015a42c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(pipeline.scheduler.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