Text Generation
Transformers
Safetensors
minimax_m2
conversational
custom_code
fp8
xuebi commited on
Commit
3df78b4
·
1 Parent(s): 8f96eee

fix: import Unpack from processing_utils

Browse files

Signed-off-by: xuebi <xuebi@minimaxi.com>

Files changed (1) hide show
  1. modeling_minimax_m2.py +2 -1
modeling_minimax_m2.py CHANGED
@@ -21,7 +21,7 @@
21
 
22
 
23
  from collections.abc import Callable
24
- from typing import Optional, Union, Unpack
25
 
26
  import torch
27
  from torch import nn
@@ -41,6 +41,7 @@ from transformers.modeling_layers import (
41
  from transformers.modeling_outputs import MoeCausalLMOutputWithPast, MoeModelOutputWithPast
42
  from transformers.modeling_rope_utils import ROPE_INIT_FUNCTIONS, dynamic_rope_update
43
  from transformers.modeling_utils import ALL_ATTENTION_FUNCTIONS, PreTrainedModel
 
44
  from transformers.utils import TransformersKwargs, auto_docstring, can_return_tuple
45
  from transformers.utils.deprecation import deprecate_kwarg
46
  from transformers.utils.generic import OutputRecorder, check_model_inputs
 
21
 
22
 
23
  from collections.abc import Callable
24
+ from typing import Optional, Union
25
 
26
  import torch
27
  from torch import nn
 
41
  from transformers.modeling_outputs import MoeCausalLMOutputWithPast, MoeModelOutputWithPast
42
  from transformers.modeling_rope_utils import ROPE_INIT_FUNCTIONS, dynamic_rope_update
43
  from transformers.modeling_utils import ALL_ATTENTION_FUNCTIONS, PreTrainedModel
44
+ from ...processing_utils import Unpack
45
  from transformers.utils import TransformersKwargs, auto_docstring, can_return_tuple
46
  from transformers.utils.deprecation import deprecate_kwarg
47
  from transformers.utils.generic import OutputRecorder, check_model_inputs