Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
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
|