diff --git a/app.py b/app.py
index 288a045..06bb56c 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)