amazonaws-la commited on
Commit
618d1f6
1 Parent(s): ad6bbc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ def generate(
56
  ) -> PIL.Image.Image:
57
  if torch.cuda.is_available():
58
  pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
59
- pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
60
 
61
  if use_lora:
62
  pipe.load_lora_weights(lora)
 
56
  ) -> PIL.Image.Image:
57
  if torch.cuda.is_available():
58
  pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
59
+ pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
60
 
61
  if use_lora:
62
  pipe.load_lora_weights(lora)