22

app.py

@@ -27,8 +27,6 @@ weight_dtype = torch.bfloat16
 transformer = ZImageControlTransformer2DModel.from_pretrained(
     MODEL_LOCAL,
     subfolder="transformer",
-    # low_cpu_mem_usage=False,
-    # torch_dtype=torch.bfloat16,
     transformer_additional_kwargs={
         "control_layers_places": [0, 5, 10, 15, 20, 25],
         "control_in_dim": 16
@@ -236,9 +234,9 @@ with gr.Blocks(css=css) as demo:
                     num_inference_steps = gr.Slider(
                         label="Number of inference steps",
                         minimum=1,
-                        maximum=30,
+                        maximum=50,
                         step=1,
-                        value=8,
+                        value=15,
                     )
         gr.Examples(examples=examples, inputs=[input_image, prompt, control_mode])
         gr.HTML(read_file("static/footer.html"))

static/data.json

@@ -1,9 +1,9 @@
 {
 	"examples": [
 		["examples/pose1.jpg", "A female paladin.", "Pose"],
-		["examples/bird.png", "A bird sitting on a branch", "Canny"],
-		["examples/bottle.png", "A man holding glass", "HED"],
-		["examples/room.png", "modern architecture, living room", "Depth"],
+		["examples/bird.jpg", "A bird sitting on a branch", "Canny"],
+		["examples/bottle.jpg", "A man holding glass", "HED"],
+		["examples/room.jpg", "modern architecture, living room", "Depth"],
 		["examples/pose2.jpg", "A fit, athletic young woman. beach background", "Pose"]
 	]
 }

static/header.html

@@ -8,6 +8,6 @@
             <br>
             Supports multiple control conditions - including Canny, HED, Depth, Pose and MLSD.
             <br>
-            If you like my spaces, please support me by visiting <a href="https://aisudo.com/" target="_blank">AiSudo</a> 😊
+            <!-- If you like my spaces, please support me by visiting <a href="https://aisudo.com/" target="_blank">AiSudo</a> 😊 -->
     </div>
 </div>