22

app.py

@@ -27,10 +27,10 @@ weight_dtype = torch.bfloat16
 transformer = ZImageControlTransformer2DModel.from_pretrained(
     MODEL_LOCAL,
     subfolder="transformer",
-    transformer_additional_kwargs={
-        "control_layers_places": [0, 5, 10, 15, 20, 25],
-        "control_in_dim": 16
-    },
+    # transformer_additional_kwargs={
+    #     "control_layers_places": [0, 5, 10, 15, 20, 25],
+    #     "control_in_dim": 16
+    # },
 ).to("cuda", torch.bfloat16)
 
 if TRANSFORMER_LOCAL is not None:
@@ -175,7 +175,7 @@ with gr.Blocks(css=css) as demo:
                 is_polish_prompt = gr.Checkbox(label="Polish prompt", value=True)
                 prompt = gr.Textbox(
                     label="Prompt",
-                    show_label=False,
+                    # show_label=False,
                     lines=2,
                     placeholder="Enter your prompt",
                     container=False,