o
    iB                     @   s$  d Z ddlZdgZddddddd	d
dddddZdddddddddddddZee Zeee  Zde	de	fdd Z
g d!Zd"d#d$d%d&d'd(d)d*d+d,
ZeZd-Zd.Zejd/ejd0d1Zd2ZdZd3Zd4Zd5Zd6Zd.Zd7Zd6ZdZd8d9d.d.d:d:d;d;d:dd<
Zd5Zd=Zd>Z d?Z!d@Z"dAZ#dBZ$dS )Cz
Spark TTS Constants

Token IDs, special tokens, speakers, and emotions for Spark TTS model.
Migrated from Veena3/Orpheus to Spark TTS (BiCodec) architecture.
    Nz
<|im_end|>                        	   
      )lipakshivardanreetNandinikrishnaanikaadarshNilayAarviAshaBittuMirar   r   r   r   r   r   r   r   r   r   r   r   )MitraAaranyaTaruNeerDhruvaIraVedaAriar   r   r   r   namereturnc                 C   s6   | t v rt |  S | tv r| S td|  ddt )a(  
    Resolve friendly speaker name to internal name.
    
    Args:
        name: Speaker name (friendly or internal)
    
    Returns:
        Internal speaker name
    
    Examples:
        resolve_speaker_name("Mitra") -> "lipakshi"
        resolve_speaker_name("lipakshi") -> "lipakshi"
    zInvalid speaker name: z. Valid names: z, )FRIENDLY_SPEAKER_MAPINDIC_SPEAKERS
ValueErrorjoinALL_SPEAKER_NAMES)r!    r(   5/home/ubuntu/veenaModal/veena3modal/core/constants.pyresolve_speaker_name7   s
   r*   )
[angry]	[curious]	[excited][giggle][laughs harder][laughs]	[screams][sighs][sings]
[whispers]r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   )
z<angry>z	<curious>z	<excited>z<giggle>z<laugh_harder>z<laugh>z<scream>z<sigh>z<sing>z	<whisper>zBayAreaBoys/spark_tts_4speakeri   
MODEL_PATHSPARK_TTS_MODEL_PATHz/models/spark_tts_4speakeri>     g?2   g      ?   bfloat16g      ?TF)
dtypegpu_memory_utilizationmax_model_lenmax_num_batched_tokensenable_prefix_cachingenable_chunked_prefillenforce_eagerdisable_log_statstrust_remote_codetensor_parallel_size@      i i i
 i	e )%__doc__osTRAINING_STOP_TOKEN_IDSSPEAKER_MAPr#   listkeysr$   r'   strr*   INDIC_EMOTION_TAGSLEGACY_EMOTION_MAPALL_EMOTION_TAGSDEFAULT_MODEL_PATHDEFAULT_MAX_MODEL_LENenvirongetBICODEC_TOKENIZER_PATHAUDIO_SAMPLE_RATEAUDIO_CHANNELSAUDIO_BITS_PER_SAMPLEDEFAULT_TEMPERATUREDEFAULT_TOP_KDEFAULT_TOP_PDEFAULT_MAX_TOKENSDEFAULT_MIN_TOKENSDEFAULT_REPETITION_PENALTYDEFAULT_SEEDVLLM_CONFIGSTREAM_BUFFER_SIZEBICODEC_BATCH_SIZEBICODEC_BATCH_TIMEOUT_MSCODE_START_TOKEN_IDCODE_END_TOKEN_IDSNAC_MIN_IDSNAC_MAX_IDr(   r(   r(   r)   <module>   s    