Merge branch 'kijai:main' into main
This commit is contained in:
@@ -17,10 +17,14 @@ from unittest.mock import patch
|
||||
from transformers.dynamic_module_utils import get_imports
|
||||
|
||||
def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
|
||||
if not str(filename).endswith("modeling_florence2.py"):
|
||||
return get_imports(filename)
|
||||
imports = get_imports(filename)
|
||||
imports.remove("flash_attn")
|
||||
try:
|
||||
if not str(filename).endswith("modeling_florence2.py"):
|
||||
return get_imports(filename)
|
||||
imports = get_imports(filename)
|
||||
imports.remove("flash_attn")
|
||||
except:
|
||||
print(f"No flash_attn import to remove")
|
||||
pass
|
||||
return imports
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user