Spaces:

sahirp
/

text_2_segment_any_vid

Sleeping

er1t0 commited on Aug 3

Commit

7976ee8

•

1 Parent(s): 5680f2a

flash attn fix

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers.dynamic_module_utils import get_imports
 def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
     """Workaround for flash_attn import issue."""
-    if not str(filename).endswith(("modeling_phi.py", "configuration_florence2.py")):
         return get_imports(filename)
     imports = get_imports(filename)
     if "flash_attn" in imports:

 def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
     """Workaround for flash_attn import issue."""
+    if not str(filename).endswith(("/modeling_florence2.py", "configuration_florence2.py")):
         return get_imports(filename)
     imports = get_imports(filename)
     if "flash_attn" in imports: