Upload 2 files

app.py

@@ -44,7 +44,7 @@ def change_base_model(repo_id: str, cn_on: bool, progress=gr.Progress(track_tqdm
             controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union_repo, torch_dtype=torch.bfloat16)
             controlnet = FluxMultiControlNetModel([controlnet_union])
             pipe = FluxControlNetPipeline.from_pretrained(repo_id, controlnet=controlnet, torch_dtype=torch.bfloat16)
-            pipe.enable_model_cpu_offload()
+            #pipe.enable_model_cpu_offload()
             last_model = repo_id
             last_cn_on = cn_on
             progress(1, desc=f"Model loaded: {repo_id} / ControlNet Loaded: {controlnet_model_union_repo}")
@@ -54,7 +54,7 @@ def change_base_model(repo_id: str, cn_on: bool, progress=gr.Progress(track_tqdm
             print(f"Loading model: {repo_id}")
             clear_cache()
             pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.bfloat16)
-            pipe.enable_model_cpu_offload()
+            #pipe.enable_model_cpu_offload()
             last_model = repo_id
             last_cn_on = cn_on
             progress(1, desc=f"Model loaded: {repo_id}")
@@ -321,7 +321,7 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=css) as app:
                 custom_lora_info = gr.HTML(visible=False)
                 custom_lora_button = gr.Button("Remove custom LoRA", visible=False)
             with gr.Column(scale=4):
-                result = gr.Image(label="Generated Image")
+                result = gr.Image(label="Generated Image", format="png", show_share_button=False)
 
         with gr.Row():
             with gr.Accordion("Advanced Settings", open=False):
@@ -394,7 +394,7 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=css) as app:
                                         cn_scale[i] = gr.Slider(label=f"ControlNet {int(i+1)} Weight", minimum=0.0, maximum=1.0, step=0.01, value=0.75)
                                         cn_res[i] = gr.Slider(label=f"ControlNet {int(i+1)} Preprocess resolution", minimum=128, maximum=512, value=384, step=1)
                                         cn_num[i] = gr.Number(i, visible=False)
-                                    cn_image[i] = gr.Image(type="pil", label="Control Image", height=256)
+                                    cn_image[i] = gr.Image(type="pil", label="Control Image", height=256, show_share_button=False)
 
     gallery.select(
         update_selection,

mod.py

@@ -148,7 +148,7 @@ def preprocess_image(image: Image.Image, control_mode: str, height: int, width:
                      preprocess_resolution: int, progress=gr.Progress(track_tqdm=True)):
     if control_mode == "None": return image
     image_resolution = max(width, height)
-    image_before = resize_image(expand2square(image), image_resolution, image_resolution, False)
+    image_before = resize_image(expand2square(image.convert("RGB")), image_resolution, image_resolution, False)
     # generated control_
     print("start to generate control image")
     preprocessor = Preprocessor()