Update optimized.py

optimized.py

@@ -99,8 +99,8 @@ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_
         controlnet_conditioning_scale=controlnet_conditioning_scale,
         num_inference_steps=steps,
         guidance_scale=guidance_scale,
-        height=control_image.size[1],
-        width=control_image.size[0]
+        height=h,
+        width=w
     ).images[0]
     print(f"VRAM used: {torch.cuda.memory_allocated()/1e9:.2f}GB")
     # Aggressive memory cleanup