Spaces:

AiSudo
/

ZIT-Controlnet

Running on Zero

Alexander Bagus commited on 11 days ago

Commit

78f31af

1 Parent(s): b48d565

22

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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
-    # },
 ).to("cuda", torch.bfloat16)
 if TRANSFORMER_LOCAL is not None:

 transformer = ZImageControlTransformer2DModel.from_pretrained(
     MODEL_LOCAL,
     subfolder="transformer",
+    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: