HuiZhang0812 commited on
Commit
fc89336
·
verified ·
1 Parent(s): 9515c64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="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")
 
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")