Fix flux memory handling issue.

pipe.enable_model_cpu_offload() vs manual pipe.to

app.py

@@ -403,7 +403,7 @@ def set_pipeline(
     pbar.close()
 
 
-@spaces.GPU(duration=200, progress=gr.Progress(track_tqdm=True))
+@spaces.GPU(duration=230, progress=gr.Progress(track_tqdm=True))
 def generate_image_lowmem(
     text,
     neg_prompt=None,
@@ -431,17 +431,18 @@ def generate_image_lowmem(
             #Enable xFormers memory-efficient attention (optional)
             #pipe.enable_xformers_memory_efficient_attention()
             print("\nEnabled xFormers memory-efficient attention.\n")
-        else:            
+        else:
             pipe.attn_implementation="flash_attention_2"
             print("\nEnabled flash_attention_2.\n")
         # alternative version that may be more efficient
         # pipe.enable_sequential_cpu_offload()
         if pipeline_name == "FluxPipeline":
             pipe.enable_model_cpu_offload()
-            pipe.vae.enable_slicing()
-            #pipe.vae.enable_tiling()
+            #pipe.vae.enable_slicing()
+            pipe.vae.enable_tiling()
         else:
-            pipe.enable_model_cpu_offload()
+            #pipe.enable_model_cpu_offload()
+            pipe.vae.enable_tiling()
 
 
         mask_parameters = {}
@@ -478,7 +479,7 @@ def generate_image_lowmem(
                     "image": conditioned_image,
                 }
             else:
-                additional_parameters ={                 
+                additional_parameters ={
                     "image": conditioned_image,
                 }
                 additional_parameters.update(mask_parameters)
@@ -526,7 +527,7 @@ def generate_image_lowmem(
     #del conditions
     del generator
     # Move the pipeline and clear cache
-    #pipe.to("cpu")
+    pipe.to("cpu")
     torch.cuda.empty_cache()
     torch.cuda.ipc_collect()
     print(torch.cuda.memory_summary(device=None, abbreviated=False))
@@ -780,7 +781,7 @@ def composite_with_control_sync(input_image, new_image, slider_value):
     
     # Load the images using open_image() if they are provided as file paths.
     new_img_path, _ = get_image_from_dict(new_image)
-    if input_image is None:
+    if (input_image is None) or (input_image["composite"] is None):
         return new_img_path
     in_img = open_image(input_image) if isinstance(input_image, (dict,str)) else input_image
     new_img = open_image(new_img_path)
@@ -1330,7 +1331,7 @@ with gr.Blocks(css_paths="style_20250314.css", title=title, theme='Surn/beeuty',
                     hura_alpha_composite_slider = gr.Slider(0,100,50,0.5, label="HURA Transparancy", elem_id="hura_alpha_composite_slider", interactive=True)
                 with gr.Row():
                     hura_button = gr.Button("Composite Input Image", elem_classes="solid")
-                    hura_sketch_button = gr.Button("Composite Sketh Image", elem_classes="solid")
+                    hura_sketch_button = gr.Button("Composite Sketch Image", elem_classes="solid")
 
                 hura_sketch_button.click(
                     fn=composite_with_control_sync,
@@ -1564,7 +1565,7 @@ with gr.Blocks(css_paths="style_20250314.css", title=title, theme='Surn/beeuty',
                                         "Replace Input Image with Template",
                                         elem_id="prerendered_replace_input_image_button",
                                         elem_classes="solid"
-                                    )                       
+                                    )
                             with gr.Row():
                                 with gr.Accordion("Template Images", open = False):
                                     with gr.Row():
@@ -1573,7 +1574,7 @@ with gr.Blocks(css_paths="style_20250314.css", title=title, theme='Surn/beeuty',
                                             prerendered_image_gallery = gr.Gallery(label="Image Gallery", show_label=True, value=build_prerendered_images_by_quality(3,'thumbnail'), elem_id="gallery", 
                                                 elem_classes="solid", type="filepath", columns=[3], rows=[3], preview=False ,object_fit="contain", height="auto", format="png",allow_preview=False)
                             with gr.Row():    
-                                image_guidance_stength = gr.Slider(label="Image Guidance Strength (prompt percentage)", info="applies to Input, Sketch and Template Image",minimum=0, maximum=1.0, value=0.9, step=0.01, interactive=True)
+                                image_guidance_stength = gr.Slider(label="Image Guidance Strength (prompt percentage)", info="applies to Input, Sketch and Template Image",minimum=0, maximum=1.0, value=0.92, step=0.01, interactive=True)
 
         with gr.Tab("Add Margins", id="margins") as margins_tab:
             with gr.Row():