bugfix

app.py

@@ -46,7 +46,7 @@ pipe = FluxControlNetInpaintPipeline.from_pretrained(base_model, controlnet=cont
 torch.backends.cuda.matmul.allow_tf32 = True
 pipe.vae.enable_tiling()
 pipe.vae.enable_slicing()
-pipe.enable_model_cpu_offload() # for saving memory
+# pipe.enable_model_cpu_offload() # for saving memory
 
 control_mode_ids = {
     "scribble_hed": 0,
@@ -184,16 +184,15 @@ def run_flux(
                 prompt=prompt,
                 image=image,
                 mask_image=mask,
-                control_image=control_image,
-                control_mode=control_mode,
-                controlnet_conditioning_scale=[0.85],
+                control_image=[control_image],
+                control_mode=[control_mode],
+                controlnet_conditioning_scale=[0.55],
                 width=width,
                 height=height,
                 strength=strength_slider,
                 generator=generator,
                 num_inference_steps=num_inference_steps_slider,
                 # max_sequence_length=256,
-                cross_attention_kwargs={"scale":0.5},
                 joint_attention_kwargs={"scale": lora_scale}
             ).images[0]
     progress(99, "Generate image success!")