Create __init__.py
Browse files- __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
|