Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -37,6 +37,7 @@ try:
|
|
| 37 |
from transformers.utils import import_utils
|
| 38 |
|
| 39 |
if "flash_attn" not in import_utils.PACKAGE_DISTRIBUTION_MAPPING: # type: ignore
|
|
|
|
| 40 |
import_utils.PACKAGE_DISTRIBUTION_MAPPING["flash_attn"] = "flash-attn" # type: ignore
|
| 41 |
except Exception as e:
|
| 42 |
print("⚠️ Patch skipped:", e)
|
|
|
|
| 37 |
from transformers.utils import import_utils
|
| 38 |
|
| 39 |
if "flash_attn" not in import_utils.PACKAGE_DISTRIBUTION_MAPPING: # type: ignore
|
| 40 |
+
print(import_utils.PACKAGE_DISTRIBUTION_MAPPING)
|
| 41 |
import_utils.PACKAGE_DISTRIBUTION_MAPPING["flash_attn"] = "flash-attn" # type: ignore
|
| 42 |
except Exception as e:
|
| 43 |
print("⚠️ Patch skipped:", e)
|