rupeshs commited on
Commit
f0f2ce8
·
1 Parent(s): 5a61b96

sdxl lighten

Browse files
Files changed (1) hide show
  1. frontend/webui/hf_demo.py +2 -2
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/sdxs-512-0.9-orig-vae"
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 = False
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()