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
|