Hugging Face's logo

Spaces:
mrfakename
/
Z-Image-Turbo
Running on Zero

App Files Community
5
multimodalart HF Staff commited on
Commit
35a8941
·
verified ·
1 Parent(s): 0866e2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -13,7 +13,6 @@ pipe = DiffusionPipeline.from_pretrained(
13
  pipe.to("cuda")
14
 
15
  # ======== AoTI compilation + FA3 ========
16
- pipe.transformer.set_attention_backend("_flash_3_hub")
17
  pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
18
  spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
19
 
 
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