Spaces:
Sleeping
Sleeping
updated app.py
Browse files
app.py
CHANGED
|
@@ -152,72 +152,6 @@ print(
|
|
| 152 |
print(
|
| 153 |
f"Found {len(app_settings.openvino_lcm_models)} OpenVINO LCM models in config/openvino-lcm-models.txt"
|
| 154 |
)
|
| 155 |
-
if args.gui:
|
| 156 |
-
from frontend.gui.ui import start_gui
|
| 157 |
-
|
| 158 |
-
print("Starting desktop GUI mode(Qt)")
|
| 159 |
-
start_gui(
|
| 160 |
-
[],
|
| 161 |
-
app_settings,
|
| 162 |
-
)
|
| 163 |
-
elif args.webui:
|
| 164 |
-
from frontend.webui.ui import start_webui
|
| 165 |
-
|
| 166 |
-
print("Starting web UI mode")
|
| 167 |
-
start_webui(
|
| 168 |
-
app_settings,
|
| 169 |
-
args.share,
|
| 170 |
-
)
|
| 171 |
-
elif args.realtime:
|
| 172 |
-
from frontend.webui.realtime_ui import start_realtime_text_to_image
|
| 173 |
-
|
| 174 |
-
print("Starting realtime text to image(EXPERIMENTAL)")
|
| 175 |
-
start_realtime_text_to_image(args.share)
|
| 176 |
-
else:
|
| 177 |
-
context = Context(InterfaceType.CLI)
|
| 178 |
-
config = app_settings.settings
|
| 179 |
-
|
| 180 |
-
if args.use_openvino:
|
| 181 |
-
config.lcm_diffusion_setting.lcm_model_id = LCM_DEFAULT_MODEL_OPENVINO
|
| 182 |
-
else:
|
| 183 |
-
config.lcm_diffusion_setting.lcm_model_id = args.lcm_model_id
|
| 184 |
-
|
| 185 |
-
config.lcm_diffusion_setting.prompt = args.prompt
|
| 186 |
-
config.lcm_diffusion_setting.image_height = args.image_height
|
| 187 |
-
config.lcm_diffusion_setting.image_width = args.image_width
|
| 188 |
-
config.lcm_diffusion_setting.guidance_scale = args.guidance_scale
|
| 189 |
-
config.lcm_diffusion_setting.number_of_images = args.number_of_images
|
| 190 |
-
config.lcm_diffusion_setting.seed = args.seed
|
| 191 |
-
config.lcm_diffusion_setting.use_openvino = args.use_openvino
|
| 192 |
-
config.lcm_diffusion_setting.use_tiny_auto_encoder = args.use_tiny_auto_encoder
|
| 193 |
-
config.lcm_diffusion_setting.use_lcm_lora = args.use_lcm_lora
|
| 194 |
-
config.lcm_diffusion_setting.lcm_lora.base_model_id = args.base_model_id
|
| 195 |
-
config.lcm_diffusion_setting.lcm_lora.lcm_lora_id = args.lcm_lora_id
|
| 196 |
-
|
| 197 |
-
if args.seed > -1:
|
| 198 |
-
config.lcm_diffusion_setting.use_seed = True
|
| 199 |
-
else:
|
| 200 |
-
config.lcm_diffusion_setting.use_seed = False
|
| 201 |
-
config.lcm_diffusion_setting.use_offline_model = args.use_offline_model
|
| 202 |
-
config.lcm_diffusion_setting.use_safety_checker = args.use_safety_checker
|
| 203 |
-
|
| 204 |
-
if args.interactive:
|
| 205 |
-
while True:
|
| 206 |
-
user_input = input(">>")
|
| 207 |
-
if user_input == "exit":
|
| 208 |
-
break
|
| 209 |
-
config.lcm_diffusion_setting.prompt = user_input
|
| 210 |
-
context.generate_text_to_image(
|
| 211 |
-
settings=config,
|
| 212 |
-
device=DEVICE,
|
| 213 |
-
)
|
| 214 |
-
|
| 215 |
-
else:
|
| 216 |
-
context.generate_text_to_image(
|
| 217 |
-
settings=config,
|
| 218 |
-
device=DEVICE,
|
| 219 |
-
)
|
| 220 |
-
|
| 221 |
|
| 222 |
from frontend.webui.hf_demo import start_demo_text_to_image
|
| 223 |
|
|
|
|
| 152 |
print(
|
| 153 |
f"Found {len(app_settings.openvino_lcm_models)} OpenVINO LCM models in config/openvino-lcm-models.txt"
|
| 154 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 155 |
|
| 156 |
from frontend.webui.hf_demo import start_demo_text_to_image
|
| 157 |
|