amazonaws-la commited on
Commit
0322008
1 Parent(s): cbcf8c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -61,12 +61,10 @@ def generate(
61
 
62
  if not use_vae:
63
  pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
64
- pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
65
 
66
  if use_vae:
67
  vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
68
  pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
69
- pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
70
 
71
  if use_lora:
72
  pipe.load_lora_weights(lora)
 
61
 
62
  if not use_vae:
63
  pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
 
64
 
65
  if use_vae:
66
  vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
67
  pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
 
68
 
69
  if use_lora:
70
  pipe.load_lora_weights(lora)