Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -22,9 +22,9 @@ ckpt_path = "HuiZhang0812/CreatiLayout"
|
|
22 |
transformer_additional_kwargs = dict(attention_type="layout",strict=True)
|
23 |
|
24 |
transformer = SiamLayoutSD3Transformer2DModel.from_pretrained(
|
25 |
-
ckpt_path, subfolder="transformer", torch_dtype=torch.
|
26 |
|
27 |
-
pipe = CreatiLayoutSD3Pipeline.from_pretrained(model_path, transformer=transformer, torch_dtype=torch.
|
28 |
pipe = pipe.to("cuda")
|
29 |
|
30 |
print("pipeline is loaded.")
|
|
|
22 |
transformer_additional_kwargs = dict(attention_type="layout",strict=True)
|
23 |
|
24 |
transformer = SiamLayoutSD3Transformer2DModel.from_pretrained(
|
25 |
+
ckpt_path, subfolder="transformer", 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")
|
29 |
|
30 |
print("pipeline is loaded.")
|