Futuretop commited on
Commit
eb50505
·
verified ·
1 Parent(s): 1042f70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -24,8 +24,9 @@ from transformers.utils import is_flash_attn_2_available
24
  _import_utils._is_package_available = lambda pkg: False
25
  _import_utils.is_flash_attn_2_available = lambda: False
26
 
27
- transformers.utils.is_flash_attn_2_available = getattr(transformers.utils, "is_flash_attn_2_available", lambda: False)
28
- transformers.utils.is_flash_attn_greater_or_equal_2_10 = lambda *args, **kwargs: False
 
29
 
30
  REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
31
 
 
24
  _import_utils._is_package_available = lambda pkg: False
25
  _import_utils.is_flash_attn_2_available = lambda: False
26
 
27
+ hf_utils = importlib.import_module('transformers.utils')
28
+ hf_utils.is_flash_attn_2_available = lambda *a, **k: False
29
+ hf_utils.is_flash_attn_greater_or_equal_2_10 = lambda *a, **k: False
30
 
31
  REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
32