Update hymm_sp/modules/models_audio.py

hymm_sp/modules/models_audio.py

@@ -6,7 +6,13 @@ import torch.nn as nn
 import torch.nn.functional as F
 from diffusers.models import ModelMixin
 from diffusers.configuration_utils import ConfigMixin, register_to_config
-from flash_attn.flash_attn_interface import flash_attn_varlen_func
+try:
+    from flash_attn.flash_attn_interface import flash_attn_varlen_func
+except ImportError:
+    print("⚠️ flash_attn not available — using fallback.")
+    flash_attn_varlen_func = None  # or create a dummy function if needed
+
+
 
 from .activation_layers import get_activation_layer
 from .norm_layers import get_norm_layer
@@ -169,6 +175,7 @@ class DoubleStreamBlock(nn.Module):
                 x.view(x.shape[0] * x.shape[1], *x.shape[2:])
                 for x in [q, k, v]
             ]
+            
             attn = flash_attn_varlen_func(
                 q,
                 k,