MindLabUnimib commited on
Commit
668feeb
·
verified ·
1 Parent(s): 2c04d38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)