rupeshs commited on
Commit
8b5373d
1 Parent(s): ea39d32

enabled tinyautoencoder

Browse files
Files changed (1) hide show
  1. frontend/webui/hf_demo.py +1 -0
frontend/webui/hf_demo.py CHANGED
@@ -42,6 +42,7 @@ def predict(
42
  use_openvino=True,
43
  use_lora=False,
44
  lcm_lora=lcm_lora,
 
45
  )
46
  print(f"prompt - {prompt}")
47
  lcm_diffusion_setting = LCMDiffusionSetting()
 
42
  use_openvino=True,
43
  use_lora=False,
44
  lcm_lora=lcm_lora,
45
+ use_tiny_auto_encoder=True,
46
  )
47
  print(f"prompt - {prompt}")
48
  lcm_diffusion_setting = LCMDiffusionSetting()