Futuretop commited on
Commit
8233d31
·
verified ·
1 Parent(s): fe12861

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ mod.__spec__ = spec
6
  sys.modules['flash_attn'] = mod
7
 
8
  import transformers.utils.import_utils as _import_utils
 
9
  _import_utils._is_package_available = lambda pkg: False
10
  _import_utils.is_flash_attn_2_available = lambda: False
11
 
 
6
  sys.modules['flash_attn'] = mod
7
 
8
  import transformers.utils.import_utils as _import_utils
9
+ from transformers.utils import is_flash_attn_2_available
10
  _import_utils._is_package_available = lambda pkg: False
11
  _import_utils.is_flash_attn_2_available = lambda: False
12