Update utils/imports.py

utils/imports.py

@@ -1,13 +1,13 @@
-import os
-
-from typing import Union
-from transformers.dynamic_module_utils import get_imports
-
-
-def fixed_get_imports(filename: Union[str, os.PathLike]) -> list[str]:
-    """Work around for https://huggingface.co/microsoft/phi-1_5/discussions/72."""
-    if not str(filename).endswith("/modeling_florence2.py"):
-        return get_imports(filename)
-    imports = get_imports(filename)
-    imports.remove("flash_attn")
+import os
+
+from typing import Union
+from transformers.dynamic_module_utils import get_imports
+
+
+def fixed_get_imports(filename: Union[str, os.PathLike]) -> list[str]:
+    if not str(filename).endswith("/modeling_florence2.py"):
+        return get_imports(filename)
+    imports = get_imports(filename)
+    if "flash_attn" in imports:
+        imports.remove("flash_attn")
     return imports