Gausson commited on
Commit
07faee0
·
verified ·
1 Parent(s): 5c2aa79

Update custom_generate/generate.py

Browse files
Files changed (1) hide show
  1. custom_generate/generate.py +1 -1
custom_generate/generate.py CHANGED
@@ -34,7 +34,7 @@ try:
34
  from .functions_2_patch import _validate_model_kwargs, llama_atten_forward
35
  from .monkey_patching_utils import monkey_patching
36
  from .sep_cache_utils import SepCache
37
- except ImportError as e:
38
  from ..functions_2_patch import _validate_model_kwargs, llama_atten_forward
39
  from ..monkey_patching_utils import monkey_patching
40
  from ..sep_cache_utils import SepCache
 
34
  from .functions_2_patch import _validate_model_kwargs, llama_atten_forward
35
  from .monkey_patching_utils import monkey_patching
36
  from .sep_cache_utils import SepCache
37
+ except :
38
  from ..functions_2_patch import _validate_model_kwargs, llama_atten_forward
39
  from ..monkey_patching_utils import monkey_patching
40
  from ..sep_cache_utils import SepCache