| from transformers import AutoConfig, AutoTokenizer, AutoModel, AutoModelForCausalLM | |
| from .modeling_qwen2 import Qwen2MTPSenseVoiceForCausalLM | |
| from .configuration_qwen2 import Qwen2MTPSenseVoiceConfig | |
| AutoConfig.register("qwen2_mtp_sensevoice", Qwen2MTPSenseVoiceConfig) | |
| AutoModelForCausalLM.register(Qwen2MTPSenseVoiceConfig, Qwen2MTPSenseVoiceForCausalLM) | |
| # AutoTokenizer.register(Qwen2MTPSenseVoiceConfig, Qwen2MTPSenseVoiceTokenizer) | |
| Qwen2MTPSenseVoiceConfig.register_for_auto_class() | |
| # Qwen2MTPSenseVoiceModel.register_for_auto_class("AutoModel") | |
| Qwen2MTPSenseVoiceForCausalLM.register_for_auto_class("AutoModelForCausalLM") | |