from transformers import DeepseekV3ForCausalLM
from .configuration_kimi_k2 import KimiK2Config


class KimiK2ForCausalLM(DeepseekV3ForCausalLM):
    config_class = KimiK2Config