fix GenerationMixin ImportError in transformers main branch

config.json

@@ -19,7 +19,8 @@
     "merge_kernel_size": [
       2,
       2
-    ]
+    ],
+    "torch_dtype": "bfloat16"
   },
   "text_config": {
     "vocab_size": 163840,

modeling_kimi_vl.py

@@ -55,10 +55,8 @@ import torch.distributed as dist
 from torch.nn import CrossEntropyLoss
 from transformers.activations import GELUActivation, ACT2FN, PytorchGELUTanh
 from transformers.cache_utils import Cache, DynamicCache
-from transformers.modeling_utils import (
-    PreTrainedModel,
-    GenerationMixin,
-)
+from transformers.modeling_utils import PreTrainedModel
+from transformers.generation.utils import GenerationMixin
 from transformers.models.llava.modeling_llava import LlavaCausalLMOutputWithPast
 from transformers.modeling_attn_mask_utils import _prepare_4d_causal_attention_mask
 from transformers.modeling_outputs import (
@@ -76,7 +74,7 @@ from transformers.utils import (
 )
 from transformers.utils.import_utils import is_torch_fx_available
 
-from .configuration_kimi_vl import MoonViTConfig, DeepseekV3Config, KimiVLConfig
+from configuration_kimi_vl import MoonViTConfig, DeepseekV3Config, KimiVLConfig
 
 
 if is_flash_attn_2_available():