internal

app.py

@@ -1,11 +1,4 @@
 import spaces
-import gradio as gr
-import argparse
-from ovi.ovi_fusion_engine import OviFusionEngine, DEFAULT_CONFIG
-from diffusers import FluxPipeline
-import tempfile
-from ovi.utils.io_utils import save_video
-from ovi.utils.processing_utils import clean_text, scale_hw_to_area_divisible
 from huggingface_hub import snapshot_download, hf_hub_download
 import os
 import subprocess
@@ -46,6 +39,23 @@ import torch
 print(f"Torch version: {torch.__version__}")
 print(f"FlashAttention available: {flash_attention_installed}")
 
+
+try:
+    import flash_attn_interface
+    FLASH_ATTN_3_AVAILABLE = True
+    print(f'FLASH_ATTN_3_AVAILABLE-innit:{FLASH_ATTN_3_AVAILABLE}')
+except ModuleNotFoundError:
+    print(f'faield FLASH_ATTN_3_AVAILABLE-innit:{FLASH_ATTN_3_AVAILABLE}')
+    FLASH_ATTN_3_AVAILABLE = False
+
+import gradio as gr
+import argparse
+from ovi.ovi_fusion_engine import OviFusionEngine, DEFAULT_CONFIG
+from diffusers import FluxPipeline
+import tempfile
+from ovi.utils.io_utils import save_video
+from ovi.utils.processing_utils import clean_text, scale_hw_to_area_divisible
+
 # ----------------------------
 # Parse CLI Args
 # ----------------------------

ovi/modules/attention.py

@@ -6,6 +6,7 @@ try:
     FLASH_ATTN_3_AVAILABLE = True
     print(f'FLASH_ATTN_3_AVAILABLE:{FLASH_ATTN_3_AVAILABLE}')
 except ModuleNotFoundError:
+    print(f'faield FLASH_ATTN_3_AVAILABLE:{FLASH_ATTN_3_AVAILABLE}')
     FLASH_ATTN_3_AVAILABLE = False
 
 try: