shenyunhang commited on
Commit
6e89714
·
verified ·
1 Parent(s): 138450a

Create __init__.py

Browse files
Files changed (1) hide show
  1. __init__.py +13 -0
__init__.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import AutoConfig, AutoTokenizer, AutoModel, AutoModelForCausalLM
2
+
3
+ from .modeling_qwen2 import Qwen2MTPSenseVoiceForCausalLM
4
+ from .configuration_qwen2 import Qwen2MTPSenseVoiceConfig
5
+
6
+ AutoConfig.register("qwen2_mtp_sensevoice", Qwen2MTPSenseVoiceConfig)
7
+ AutoModelForCausalLM.register(Qwen2MTPSenseVoiceConfig, Qwen2MTPSenseVoiceForCausalLM)
8
+ # AutoTokenizer.register(Qwen2MTPSenseVoiceConfig, Qwen2MTPSenseVoiceTokenizer)
9
+
10
+ Qwen2MTPSenseVoiceConfig.register_for_auto_class()
11
+ # Qwen2MTPSenseVoiceModel.register_for_auto_class("AutoModel")
12
+ Qwen2MTPSenseVoiceForCausalLM.register_for_auto_class("AutoModelForCausalLM")
13
+ VertiTab