刘虹雨 commited on
Commit
2f1e1a3
·
1 Parent(s): c8bb017

update code

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1056,7 +1056,7 @@ if __name__ == '__main__':
1056
  sd_path, torch_dtype=torch.float16, text_encoder=text_encoder,
1057
  use_safetensors=True, controlnet=controlnet, variant="fp16"
1058
  ).to(device)
1059
- pipeline_sd.scheduler=DPMSolverMultistepScheduler.from_config(pipeline_sd.scheduler_config, use_karras_sigmas=True)
1060
 
1061
  demo_cam = False
1062
  base_coff = np.load(
 
1056
  sd_path, torch_dtype=torch.float16, text_encoder=text_encoder,
1057
  use_safetensors=True, controlnet=controlnet, variant="fp16"
1058
  ).to(device)
1059
+ pipeline_sd.scheduler=DPMSolverMultistepScheduler.from_config(pipeline_sd.scheduler.config, use_karras_sigmas=True)
1060
 
1061
  demo_cam = False
1062
  base_coff = np.load(