From a7f32912f5c6556a3a562d98c476e16b4f722128 Mon Sep 17 00:00:00 2001
From: Shenghai Yuan <140951558+SHYuanBest@users.noreply.github.com>
Date: Fri, 6 Mar 2026 16:55:28 +0800
Subject: [PATCH] Update app.py

---
 app.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/app.py b/app.py
index 4afc549..0fa6349 100644
--- a/app.py
+++ b/app.py
@@ -31,8 +31,10 @@ pipe = HeliosPyramidPipeline.from_pretrained(
 )
 
 pipe.to("cuda")
-# pipe.transformer.set_attention_backend("flash_hub")
-pipe.transformer.set_attention_backend("_flash_3_hub")
+try:
+    pipe.transformer.set_attention_backend("_flash_3_hub")
+except Exception:
+    pipe.transformer.set_attention_backend("flash_hub")
 
 # @spaces.GPU(duration=1500)
 # def compile_transformer():
@@ -260,4 +262,4 @@ with gr.Blocks(css=CSS, title="Helios Video Generation", theme=gr.themes.Soft())
 
 if __name__ == "__main__":
     # demo.launch(share=True, allowed_paths=["./examples"])
-    demo.launch(share=True)
\ No newline at end of file
+    demo.launch(share=True)