fffiloni commited on
Commit
4a777a4
·
verified ·
1 Parent(s): 1d9d947

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ from unittest.mock import patch
12
  from transformers.dynamic_module_utils import get_imports
13
 
14
  def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
15
- if not str(filename).endswith("/modeling_phi3.py"):
16
- return get_imports(filename)
17
  imports = get_imports(filename)
18
  imports.remove("flash_attn")
19
  return imports
 
12
  from transformers.dynamic_module_utils import get_imports
13
 
14
  def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
15
+ #if not str(filename).endswith("/modeling_phi3.py"):
16
+ # return get_imports(filename)
17
  imports = get_imports(filename)
18
  imports.remove("flash_attn")
19
  return imports