yingzhac-research commited on
Commit
f89593c
·
1 Parent(s): 049b5a3

Disable AoTI+FA3 on HF runtime

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,9 +14,9 @@ pipe = DiffusionPipeline.from_pretrained(
14
  )
15
  pipe.to("cuda")
16
 
17
- # ======== AoTI compilation + FA3 ========
18
- pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
19
- spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
20
 
21
  print("Pipeline loaded!")
22
 
 
14
  )
15
  pipe.to("cuda")
16
 
17
+ '# ======== AoTI compilation + FA3 ======== (disabled on HF to avoid outdated AOTI/FA3 package errors)'
18
+ # pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
19
+ # spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
20
 
21
  print("Pipeline loaded!")
22