mrfakename commited on
Commit
e7338c1
·
verified ·
1 Parent(s): 190e9bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,8 +13,8 @@ pipe = DiffusionPipeline.from_pretrained(
13
  pipe.to("cuda")
14
 
15
  # ======== AoTI compilation + FA3 ========
16
- pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
17
- spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
18
 
19
  print("Pipeline loaded!")
20
 
 
13
  pipe.to("cuda")
14
 
15
  # ======== AoTI compilation + FA3 ========
16
+ # pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
17
+ # spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
18
 
19
  print("Pipeline loaded!")
20