Update frontend/webui/image_to_image_ui.py

frontend/webui/image_to_image_ui.py

@@ -31,17 +31,16 @@ def generate_image_to_image(
     print("新图",newH,newW)
     app_settings.settings.lcm_diffusion_setting.prompt = prompt
     app_settings.settings.lcm_diffusion_setting.negative_prompt = negative_prompt
-    app_settings.settings.lcm_diffusion_setting.init_image = img
+    app_settings.settings.lcm_diffusion_setting.init_image = init_image
     app_settings.settings.lcm_diffusion_setting.strength = strength
-    app_settings.settings.lcm_diffusion_setting.image_width=newW
-    app_settings.settings.lcm_diffusion_setting.image_height=newH
+   
     app_settings.settings.lcm_diffusion_setting.diffusion_task = (
         DiffusionTask.image_to_image.value
     )
     model_id = app_settings.settings.lcm_diffusion_setting.openvino_lcm_model_id
     reshape = False
-    image_width = newW
-    image_height =newH
+    image_width = app_settings.settings.lcm_diffusion_setting.image_width
+    image_height = app_settings.settings.lcm_diffusion_setting.image_height
     num_images = app_settings.settings.lcm_diffusion_setting.number_of_images
     if app_settings.settings.lcm_diffusion_setting.use_openvino:
         reshape = is_reshape_required(
@@ -55,6 +54,7 @@ def generate_image_to_image(
             num_images,
         )
 
+
     with ThreadPoolExecutor(max_workers=1) as executor:
         future = executor.submit(
             context.generate_text_to_image,