25 lines
733 B
Python
25 lines
733 B
Python
"""Fish Speech S2 Pro config registration with transformers AutoConfig.
|
|
|
|
Registers FishSpeechConfig (model_type="fish_qwen3_omni") and sub-configs
|
|
so that ``AutoConfig.from_pretrained("fishaudio/s2-pro")`` returns the
|
|
correct config class.
|
|
"""
|
|
|
|
from transformers import AutoConfig
|
|
|
|
from vllm_omni.model_executor.models.fish_speech.configuration_fish_speech import (
|
|
FishSpeechConfig,
|
|
FishSpeechFastARConfig,
|
|
FishSpeechSlowARConfig,
|
|
)
|
|
|
|
AutoConfig.register("fish_qwen3_omni", FishSpeechConfig)
|
|
AutoConfig.register("fish_qwen3", FishSpeechSlowARConfig)
|
|
AutoConfig.register("fish_qwen3_audio_decoder", FishSpeechFastARConfig)
|
|
|
|
__all__ = [
|
|
"FishSpeechConfig",
|
|
"FishSpeechSlowARConfig",
|
|
"FishSpeechFastARConfig",
|
|
]
|