amazonaws-la commited on
Commit
c3cd60c
·
verified ·
1 Parent(s): 5b7982f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -61,13 +61,11 @@ 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)
73
  pipe.fuse_lora(lora_scale=0.7)
 
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)
71
  pipe.fuse_lora(lora_scale=0.7)