jotase commited on
Commit
133e476
·
verified ·
1 Parent(s): f5c1620

Update utils/florence.py

Browse files
Files changed (1) hide show
  1. utils/florence.py +2 -1
utils/florence.py CHANGED
@@ -21,7 +21,8 @@ def fixed_get_imports(filename: Union[str, os.PathLike]) -> list[str]:
21
  if not str(filename).endswith("/modeling_florence2.py"):
22
  return get_imports(filename)
23
  imports = get_imports(filename)
24
- imports.remove("flash_attn")
 
25
  return imports
26
 
27
 
 
21
  if not str(filename).endswith("/modeling_florence2.py"):
22
  return get_imports(filename)
23
  imports = get_imports(filename)
24
+ if "flash_attn" in imports:
25
+ imports.remove("flash_attn")
26
  return imports
27
 
28