cbensimon HF Staff commited on
Commit
b796046
·
verified ·
1 Parent(s): d5c14d7

Remove ahead-of-time compiled version loading for now

Browse files

FlashAttention-3 is not yet available on new ZeroGPU hardware (Blackwell architecture)

Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -482,11 +482,6 @@ def generate(
482
 
483
  init_app()
484
 
485
- # ==================== AoTI (Ahead of Time Inductor compilation) ====================
486
-
487
- pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
488
- spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
489
-
490
  with gr.Blocks(title="Z-Image Demo") as demo:
491
  gr.Markdown(
492
  """<div align="center">
 
482
 
483
  init_app()
484
 
 
 
 
 
 
485
  with gr.Blocks(title="Z-Image Demo") as demo:
486
  gr.Markdown(
487
  """<div align="center">