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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.bfloat16,**transformer_additional_kwargs)
26
 
27
- pipe = CreatiLayoutSD3Pipeline.from_pretrained(model_path, transformer=transformer, torch_dtype=torch.bfloat16)
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.")