sdxl lighten
Browse files
frontend/webui/hf_demo.py
CHANGED
@@ -42,7 +42,7 @@ def predict(
|
|
42 |
print(f"prompt - {prompt}")
|
43 |
lcm_diffusion_setting = LCMDiffusionSetting()
|
44 |
lcm_diffusion_setting.diffusion_task = DiffusionTask.text_to_image.value
|
45 |
-
lcm_diffusion_setting.openvino_lcm_model_id = "rupeshs/
|
46 |
lcm_diffusion_setting.use_lcm_lora = False
|
47 |
lcm_diffusion_setting.prompt = prompt
|
48 |
lcm_diffusion_setting.guidance_scale = 1.0
|
@@ -56,7 +56,7 @@ def predict(
|
|
56 |
lcm_diffusion_setting.image_width = 512
|
57 |
lcm_diffusion_setting.image_height = 512
|
58 |
lcm_diffusion_setting.use_openvino = False
|
59 |
-
lcm_diffusion_setting.use_tiny_auto_encoder =
|
60 |
pprint(lcm_diffusion_setting.model_dump())
|
61 |
lcm_text_to_image.init(lcm_diffusion_setting=lcm_diffusion_setting)
|
62 |
start = perf_counter()
|
|
|
42 |
print(f"prompt - {prompt}")
|
43 |
lcm_diffusion_setting = LCMDiffusionSetting()
|
44 |
lcm_diffusion_setting.diffusion_task = DiffusionTask.text_to_image.value
|
45 |
+
lcm_diffusion_setting.openvino_lcm_model_id = "rupeshs/SDXL-Lightning-2steps"
|
46 |
lcm_diffusion_setting.use_lcm_lora = False
|
47 |
lcm_diffusion_setting.prompt = prompt
|
48 |
lcm_diffusion_setting.guidance_scale = 1.0
|
|
|
56 |
lcm_diffusion_setting.image_width = 512
|
57 |
lcm_diffusion_setting.image_height = 512
|
58 |
lcm_diffusion_setting.use_openvino = False
|
59 |
+
lcm_diffusion_setting.use_tiny_auto_encoder = True
|
60 |
pprint(lcm_diffusion_setting.model_dump())
|
61 |
lcm_text_to_image.init(lcm_diffusion_setting=lcm_diffusion_setting)
|
62 |
start = perf_counter()
|