Function register_funaudiochat

funaudiochat/register.py:15–30

Register FunAudioChat classes with AutoConfig and AutoModel.

()

# Registration function for explicit use
def register_funaudiochat():
    """Register FunAudioChat classes with AutoConfig and AutoModel."""
    from transformers import AutoConfig, AutoModelForSeq2SeqLM, AutoProcessor
    from .configuration_funaudiochat import FunAudioChatConfig, FunAudioChatAudioEncoderConfig
    from .modeling_funaudiochat import FunAudioChatForConditionalGeneration
    from .processing_funaudiochat import FunAudioChatProcessor
    
    # Register configurations
    AutoConfig.register("funaudiochat", FunAudioChatConfig)
    AutoConfig.register("funaudiochat_audio_encoder", FunAudioChatAudioEncoderConfig)
    
    # Register processor
    AutoProcessor.register(FunAudioChatConfig, FunAudioChatProcessor)
    
    # Register model
    AutoModelForSeq2SeqLM.register(FunAudioChatConfig, FunAudioChatForConditionalGeneration)