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
|