Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
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()
|