Update optimized.py

optimized.py

@@ -93,6 +93,7 @@ if getattr(pipe, "vae", None) is not None:
         pipe.vae.enable_slicing()
     except AttributeError:
         # Method 2: Apply manual slicing for Flux compatibility [source_id]pipeline_flux_controlnet.py
+        print("Falling back to manual attention slicing.")
         pipe.vae.decode = self_attention_slicing(pipe.vae.decode, 2) 
 
 pipe.enable_attention_slicing(1)
@@ -106,7 +107,7 @@ pipe.enable_attention_slicing(1)
 
 print(f"VRAM used: {torch.cuda.memory_allocated()/1e9:.2f}GB")
 @spaces.GPU
-def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_scale, guidance_scale):
+def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_scale, guidance_scale, guidance_start, guidance_end):
     print(f"Memory Usage: {torch.cuda.memory_summary(device=None, abbreviated=False)}")
 
     # Load control image
@@ -123,13 +124,10 @@ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_
         num_inference_steps=steps,
         guidance_scale=guidance_scale,
         height=h,
-        width=w
+        width=w,
+        control_guidance_start=control_guidance_start,
+        control_guidance_end=control_guidance_end
     ).images[0]
-    print(f"VRAM used: {torch.cuda.memory_allocated()/1e9:.2f}GB")
-    # Aggressive memory cleanup
-    # torch.cuda.empty_cache()
-    # torch.cuda.ipc_collect()
-    print(f"VRAM used: {torch.cuda.memory_allocated()/1e9:.2f}GB")
     return image
 # Create Gradio interface
 iface = gr.Interface(
@@ -139,8 +137,12 @@ iface = gr.Interface(
         gr.Slider(1, 3, value=1, label="Scale"),
         gr.Slider(2, 20, value=8, label="Steps"),
         gr.Image(type="pil", label="Control Image"),
-        gr.Slider(0, 1, value=0.6, label="ControlNet Scale"),
-        gr.Slider(1, 20, value=3.5, label="Guidance Scale"),
+        with gr.Accordion(label="Settings (Advanced)"):
+            gr.Slider(0, 1, value=0.6, label="ControlNet Scale"),
+            gr.Slider(1, 20, value=3.5, label="Guidance Scale"),
+            with gr.Row:
+                gr.Slider(0, 1, value=0.0, label="Control Guidance Start"),
+                gr.Slider(0, 1, value=1.0 label="Control Guidance End"),
     ],
     outputs=[
         gr.Image(type="pil", label="Generated Image", format="png"),