None1145 commited on
Commit
8116c4c
·
verified ·
1 Parent(s): b46a44f

Update app.py

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