potsawee commited on
Commit
b1df7b3
·
verified ·
1 Parent(s): e72ddf8

Upload modeling_mimi_clean.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_mimi_clean.py +5 -1
modeling_mimi_clean.py CHANGED
@@ -31,7 +31,11 @@ from transformers.modeling_outputs import BaseModelOutputWithPast
31
  from transformers.modeling_rope_utils import ROPE_INIT_FUNCTIONS, dynamic_rope_update
32
  from transformers.modeling_utils import PreTrainedModel
33
  from transformers.utils import ModelOutput, auto_docstring, logging
34
- from configuration_mimi import MimiConfig
 
 
 
 
35
 
36
 
37
  if is_flash_attn_available():
 
31
  from transformers.modeling_rope_utils import ROPE_INIT_FUNCTIONS, dynamic_rope_update
32
  from transformers.modeling_utils import PreTrainedModel
33
  from transformers.utils import ModelOutput, auto_docstring, logging
34
+
35
+ try:
36
+ from .configuration_mimi import MimiConfig
37
+ except ImportError:
38
+ from configuration_mimi import MimiConfig
39
 
40
 
41
  if is_flash_attn_available():