None1145 commited on
Commit
1d09186
1 Parent(s): 376d672

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -22,10 +22,10 @@ def reload_model(new_model_id):
22
  pipe = OVStableDiffusionXLPipeline.from_pretrained(model_id, compile=False)
23
  if model_id == "None1145/noobai-XL-Vpred-0.65s-openvino":
24
  scheduler_args = {"prediction_type": "v_prediction", "rescale_betas_zero_snr": True}
25
- # pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, **scheduler_args)
26
- pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config, **scheduler_args)
27
- pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl")
28
- pipe.fuse_lora()
29
  # pipe.to("gpu")
30
  pipe.reshape(batch_size=1, height=prev_height, width=prev_width, num_images_per_prompt=1)
31
  pipe.compile()
 
22
  pipe = OVStableDiffusionXLPipeline.from_pretrained(model_id, compile=False)
23
  if model_id == "None1145/noobai-XL-Vpred-0.65s-openvino":
24
  scheduler_args = {"prediction_type": "v_prediction", "rescale_betas_zero_snr": True}
25
+ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, **scheduler_args)
26
+ # pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config, **scheduler_args)
27
+ # pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl")
28
+ # pipe.fuse_lora()
29
  # pipe.to("gpu")
30
  pipe.reshape(batch_size=1, height=prev_height, width=prev_width, num_images_per_prompt=1)
31
  pipe.compile()