rupeshs commited on
Commit
736dfe5
·
1 Parent(s): 4f971c2

updated app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -66
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