Spaces:
Running
Running
added safety filter tiny vae
Browse files
frontend/webui/hf_demo.py
CHANGED
@@ -50,6 +50,8 @@ def predict(
|
|
50 |
lcm_diffusion_setting.inference_steps = steps
|
51 |
lcm_diffusion_setting.seed = seed
|
52 |
lcm_diffusion_setting.use_seed = use_seed
|
|
|
|
|
53 |
lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
|
54 |
lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
|
55 |
lcm_diffusion_setting.use_openvino = True if is_openvino_device() else False
|
|
|
50 |
lcm_diffusion_setting.inference_steps = steps
|
51 |
lcm_diffusion_setting.seed = seed
|
52 |
lcm_diffusion_setting.use_seed = use_seed
|
53 |
+
lcm_diffusion_setting.use_safety_checker = True
|
54 |
+
lcm_diffusion_setting.use_tiny_auto_encoder = True
|
55 |
lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
|
56 |
lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
|
57 |
lcm_diffusion_setting.use_openvino = True if is_openvino_device() else False
|