22

app.py

@@ -37,8 +37,10 @@ transformer = ZImageControlTransformer2DModel.from_pretrained(
     MODEL_LOCAL,
     subfolder="transformer",
     transformer_additional_kwargs={
-        "control_layers_places": [0, 5, 10, 15, 20, 25],
-        "control_in_dim": 16
+        "control_layers_places": [0, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28],
+        "control_refiner_layers_places": [0, 1],
+        "add_control_noise_refiner": True,
+        "control_in_dim": 33
     },
 ).to("cuda", torch.bfloat16)