o
    i                     @   s8   d dl mZ edddd Zedddd ZdS )    )tablestokenizer_classesWhisperTokenizerc                  K   sf   zddl m} W n   td Y | dd }| dd}| dd}| d	d
}|||||d}|S )Nr   get_tokenizerJNotice: If you want to use whisper, please `pip install -U openai-whisper`languagetask
transcribeis_multilingualTnum_languagesc   )multilingualr   r   r	   )whisper.tokenizerr   printget)kwargsr   r   r	   r   r   	tokenizer r   V/home/ubuntu/.local/lib/python3.10/site-packages/funasr/tokenizer/whisper_tokenizer.pyr      s   
SenseVoiceTokenizerc                  K   st   zddl m} W n   td Y | dd }| dd }| dd}| dd	}| d
d }||||||d}|S )Nr   r   r   r   r	   r   Tr   i-"  
vocab_path)r   r   r   r	   r   )/funasr.models.sense_voice.whisper_lib.tokenizerr   r   r   )r   r   r   r	   r   r   r   r   r   r   r   r      s"   
N)funasr.registerr   registerr   r   r   r   r   r   <module>   s
    


