Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ ckpt_path = "HuiZhang0812/CreatiLayout"
|
|
22 |
transformer_additional_kwargs = dict(attention_type="layout",strict=True)
|
23 |
|
24 |
transformer = SiamLayoutSD3Transformer2DModel.from_pretrained(
|
25 |
-
ckpt_path, subfolder="
|
26 |
|
27 |
pipe = CreatiLayoutSD3Pipeline.from_pretrained(model_path, transformer=transformer, torch_dtype=torch.float16)
|
28 |
pipe = pipe.to("cuda")
|
|
|
22 |
transformer_additional_kwargs = dict(attention_type="layout",strict=True)
|
23 |
|
24 |
transformer = SiamLayoutSD3Transformer2DModel.from_pretrained(
|
25 |
+
ckpt_path, subfolder="SiamLayout_SD3", torch_dtype=torch.float16,**transformer_additional_kwargs)
|
26 |
|
27 |
pipe = CreatiLayoutSD3Pipeline.from_pretrained(model_path, transformer=transformer, torch_dtype=torch.float16)
|
28 |
pipe = pipe.to("cuda")
|