ZhaoRuiyu commited on
Commit
f6624e4
·
verified ·
1 Parent(s): 47c0500

Update modeling_xmodel.py

Browse files
Files changed (1) hide show
  1. modeling_xmodel.py +2 -1
modeling_xmodel.py CHANGED
@@ -23,7 +23,8 @@ from transformers.modeling_outputs import (
23
  TokenClassifierOutput,
24
  )
25
  from transformers.modeling_rope_utils import ROPE_INIT_FUNCTIONS
26
- from transformers.modeling_utils import PreTrainedModel, GenerationMixin
 
27
  from transformers.pytorch_utils import ALL_LAYERNORM_LAYERS
28
  from transformers.utils import (
29
  add_start_docstrings,
 
23
  TokenClassifierOutput,
24
  )
25
  from transformers.modeling_rope_utils import ROPE_INIT_FUNCTIONS
26
+ from transformers.modeling_utils import PreTrainedModel
27
+ from transformers.generation import GenerationMixin
28
  from transformers.pytorch_utils import ALL_LAYERNORM_LAYERS
29
  from transformers.utils import (
30
  add_start_docstrings,