Futuretop commited on
Commit
de7d0eb
·
verified ·
1 Parent(s): 684e4b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,6 +1,9 @@
1
- import sys, types
2
 
3
- sys.modules['flash_attn'] = types.ModuleType('flash_attn')
 
 
 
4
 
5
  import transformers.dynamic_module_utils as dmu
6
  dmu.check_imports = lambda *args, **kwargs: []
 
1
+ import sys, types, importlib.machinery
2
 
3
+ spec = importlib.machinery.ModuleSpec('flash_attn', loader=None)
4
+ mod = types.ModuleType('flash_attn')
5
+ mod.__spec__ = spec
6
+ sys.modules['flash_attn'] = mod
7
 
8
  import transformers.dynamic_module_utils as dmu
9
  dmu.check_imports = lambda *args, **kwargs: []