Spaces:

AiSudo
/

ZIT-Controlnet

Running on Zero

Alexander Bagus commited on 7 days ago

Commit

e8db9a2

1 Parent(s): ed56a3e

22

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,8 +27,8 @@ weight_dtype = torch.bfloat16
 transformer = ZImageControlTransformer2DModel.from_pretrained(
     MODEL_LOCAL,
     subfolder="transformer",
-    low_cpu_mem_usage=True,
-    torch_dtype=torch.bfloat16,
     transformer_additional_kwargs={
         "control_layers_places": [0, 5, 10, 15, 20, 25],
         "control_in_dim": 16

 transformer = ZImageControlTransformer2DModel.from_pretrained(
     MODEL_LOCAL,
     subfolder="transformer",
+    low_cpu_mem_usage=False,
+    dtype=torch.bfloat16,
     transformer_additional_kwargs={
         "control_layers_places": [0, 5, 10, 15, 20, 25],
         "control_in_dim": 16