Spaces:

AiSudo
/

ZIT-Inpaint

Running on Zero

Alexander Bagus commited on Dec 13, 2025

Commit

d5e0780

1 Parent(s): 3f4f137

22

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,7 +34,7 @@ DTYPE = torch.bfloat16
 transformer = ZImageControlTransformer2DModel.from_pretrained(
     MODEL_LOCAL,
     subfolder="transformer",
-    transformer_additional_kwargs=OmegaConf.to_container(TRANSFORMER_CONFIG['transformer_additional_kwargs']),
 ).to("cuda", torch.bfloat16)
 state_dict = load_file(TRANSFORMER_LOCAL)

 transformer = ZImageControlTransformer2DModel.from_pretrained(
     MODEL_LOCAL,
     subfolder="transformer",
+    transformer_additional_kwargs=OmegaConf.to_container(TRANSFORMER_CONFIG)['transformer_additional_kwargs'],
 ).to("cuda", torch.bfloat16)
 state_dict = load_file(TRANSFORMER_LOCAL)