Update hymm_sp/modules/models_audio.py

hymm_sp/modules/models_audio.py

@@ -6,11 +6,9 @@ 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
-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 flash_attn.flash_attn_interface import flash_attn_varlen_func
+