PhPHdZddlmZddlmZej eZGddeZGddeZ Gdd eZ Gd d eZ Gd d eZ gdZ y)zBlt model configuration)PretrainedConfig)loggingcHeZdZdZdZ dfd ZxZS)BltLocalEncoderConfigzB Configuration class for the Blt Local Encoder component. blt_local_encoderc r||_||_||_||_||_||_|xs||_||z|_|xstd|zdz |_ ||_ | |_ | |_ | |_ | |_| |_||_||_|j%ddt'|Pdi|ddiyNrtie_word_embeddingsF vocab_sizecross_attn_all_layers cross_attn_khidden_size_global hidden_sizenum_attention_headsnum_key_value_headshead_dimintintermediate_sizenum_hidden_layers rms_norm_epsdropoutmax_position_embeddings rope_theta rope_scaling hidden_actinitializer_rangepopsuper__init__selfrrrrrrrrrrrrrrrrkwargs __class__s c/var/www/html/saasai/venv/lib/python3.12/site-packages/transformers/models/blt/configuration_blt.pyr"zBltLocalEncoderConfig.__init__(%%:"("4&#6 #6#M:M #':: !2!Nc!k/A:M6N!2( '>$$($!2  ($/ =6=u=)FNh㈵>`ANsilu {Gz?__name__ __module__ __qualname____doc__ model_typer" __classcell__r&s@r'rrsK%J#  %#(>(>r)rcHeZdZdZdZ dfd ZxZS)BltLocalDecoderConfigzB Configuration class for the Blt Local Decoder component. blt_local_decoderc r||_||_||_||_||_||_|xs||_||z|_|xstd|zdz |_ ||_ | |_ | |_ | |_ | |_| |_||_||_|j%ddt'|Pdi|ddiyr r r#s r'r"zBltLocalDecoderConfig.__init__Qr(r))r*Tr+r,r-r.N r0r1r2r3Nr4r5r6r7r>s@r'r@r@JsK%J"  %#(>(>r)r@c@eZdZdZdZ dfd ZxZS)BltGlobalTransformerConfigzG Configuration class for the Blt Global Transformer component. blt_global_transformerc  :||_||_|xs||_||z|_| xst d|zdz |_||_||_||_||_ ||_ | |_ | |_ | |_ | jddt|@di| ddiyr )rrrrrrrrrrrrrrr r!r")r$rrrrrrrrrrrrr%r&s r'r"z#BltGlobalTransformerConfig.__init__s '#6 #6#M:M #':: !2!Nc!k/A:M6N!2( '>$$($!2  ($/ =6=u=r)) r,r.Nr0r1r3Nr4ir6r7r>s@r'rErE|s?*J  $ > >r)rEc@eZdZdZdZ dfd ZxZS)BltPatcherConfiga Configuration class for the Blt Patcher/Entropy model component. Args: vocab_size (`int`, *optional*, defaults to 260): Vocabulary size of the Blt patcher model. Defines the number of different tokens that can be represented by the `inputs_ids` passed when calling the patcher model. hidden_size (`int`, *optional*, defaults to 768): Dimension of the hidden representations. num_hidden_layers (`int`, *optional*, defaults to 14): Number of hidden layers in the Transformer decoder. num_attention_heads (`int`, *optional*, defaults to 12): Number of attention heads for each attention layer in the Transformer decoder. num_key_value_heads (`int`, *optional*): This is the number of key_value heads that should be used to implement Grouped Query Attention. If `num_key_value_heads=num_attention_heads`, the model will use Multi Head Attention (MHA), if `num_key_value_heads=1` the model will use Multi Query Attention (MQA) otherwise GQA is used. When converting a multi-head checkpoint to a GQA checkpoint, each group key and value head should be constructed by meanpooling all the original heads within that group. For more details, check out [this paper](https://huggingface.co/papers/2305.13245). If it is not specified, will default to `num_attention_heads`. max_position_embeddings (`int`, *optional*, defaults to 8192): The maximum sequence length that this model might ever be used with. rms_norm_eps (`float`, *optional*, defaults to 1e-05): The epsilon used by the rms normalization layers. dropout (`float`, *optional*, defaults to 0.0): The dropout ratio for the attention probabilities. rope_theta (`float`, *optional*, defaults to 10000.0): The base period of the RoPE embeddings. intermediate_size (`int`, *optional*, defaults to 2048): Dimension of the MLP representations. rope_scaling (`dict`, *optional*): Dictionary containing the RoPE scaling configuration. initializer_range (`float`, *optional*, defaults to 0.02): The standard deviation of the truncated_normal_initializer for initializing all weight matrices. blt_patcherc  \||_||_||_||_||z|_||n||_||_||_||_| |_ d|_ | xstd|jzdz |_ | |_ | |_| jddt!|Ddi| ddiy)Nr4r rr Fr )rrrrrrrrrrrrrrrr r!r")r$rrrrrrrrrrrrr%r&s r'r"zBltPatcherConfig.__init__s %&!2#6 #':: :M:Y#6_r '>$( $ !2!Sc!d>N>N:NQR:R6S(!2  ($/ =6=u=r)) r*i Ni r0r1g@r,Nr6r7r>s@r'rKrKs@#JJ $!>!>r)rKcdeZdZdZdZdgZeeee dZ dfd Z xZ S) BltConfigas This is the configuration class to store the configuration of a [`BltModel`]. It is used to instantiate a Blt model according to the specified arguments, defining the model architecture. Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the documentation from [`PretrainedConfig`] for more information. Args: vocab_size (`int`, *optional*, defaults to 260): Vocabulary size of the Blt model. Defines the number of different tokens that can be represented by the `inputs_ids` passed when calling [`BltModel`]. max_position_embeddings (`int`, *optional*, defaults to 4096): The maximum sequence length that this model might ever be used with. patch_in_forward (`bool`, *optional*, defaults to `True`): Whether to perform patching during the forward pass. patch_size (`int`, *optional*, defaults to 4): Size of the patches used in the patching mechanism. patching_mode (`str`, *optional*, defaults to `"entropy"`): The mode used for patching, such as entropy-based patching. patching_threshold (`float`, *optional*, defaults to 1.34): Threshold value used for determining when to apply patches. patching_batch_size (`int`, *optional*, defaults to 1): Batch size used during the patching process. max_patch_length (`int`, *optional*): Maximum length of patches that can be generated. cross_attn_k (`int`, *optional*, defaults to 2): Number of cross-attention heads used in the model. encoder_hash_byte_group_size (`list`, *optional*): List of byte group sizes used in the encoder hash function. encoder_hash_byte_group_vocab (`int`, *optional*, defaults to 500002): Vocabulary size for the encoder hash byte groups. encoder_hash_byte_group_nb_functions (`int`, *optional*, defaults to 1): Number of hash functions used in the encoder byte grouping. patcher_config (`BltPatcherConfig`, *optional*): Configuration for the patcher component of the model. encoder_config (`BltLocalEncoderConfig`, *optional*): Configuration for the local encoder component of the model. decoder_config (`BltLocalDecoderConfig`, *optional*): Configuration for the local decoder component of the model. global_config (`BltGlobalTransformerConfig`, *optional*): Configuration for the global transformer component of the model. tie_word_embeddings (`bool`, *optional*, defaults to `False`): Whether to tie weight embeddings. initializer_range (`float`, *optional*, defaults to 0.02): The standard deviation of the truncated_normal_initializer for initializing all weight matrices. rope_theta (`float`, *optional*, defaults to 500000.0): The base period of the RoPE embeddings. rope_scaling (`dict`, *optional*): Dictionary containing the RoPE scaling configuration. ```python >>> from transformers import BltModel, BltConfig >>> # Initializing a Blt configuration >>> configuration = BltConfig() >>> # Initializing a model from the configuration >>> model = BltModel(configuration) >>> # Accessing the model configuration >>> configuration = model.config ``` Checkpoint: [facebook/blt](https://huggingface.co/facebook/blt) bltpast_key_values)patcher_configencoder_configdecoder_config global_configc ||_||_||_||_||_||_||_||_||_||_ ||_ |jdd|_ |jdd|_ |jd|_|jdd|_| |_| xsgd|_| |_| |_| 't)| |_t,j/d nJt1| t2r#| j5d |t)di| |_nt1| t(r| |_|'t7| |_t,j/d nJt1|t2r#|j5d |t7di||_nt1|t6r||_|'t;| |_t,j/d nJt1|t2r#|j5d |t;di||_nt1|t:r||_|'t?| |_ t,j/dnJt1|t2r#|j5d |t?di||_ nt1|t>r||_ |j8jB|j z}||j@jBk7r|nd|j@_"|jGddtI|dd|i|y)Npatching_devicecudarealtime_patchingTpatching_threshold_add monotonicityF)rr )rz8patcher_config is None, using default Blt patcher configrz8encoder_config is None, using default Blt encoder configz8decoder_config is None, using default Blt decoder configz6global_config is None, using default Blt global configr r )&rrrrrpatch_in_forward patch_size patching_modepatching_thresholdpatching_batch_sizemax_patch_lengthgetrYr[r\r]rencoder_hash_byte_group_sizeencoder_hash_byte_group_vocab$encoder_hash_byte_group_nb_functionsrKrTloggerinfo isinstancedict setdefaultrrUr@rVrErWrencoder_cross_output_sizer r!r")r$rrrbrcrdrerfrgrrirjrkrTrUrVrWr rrrr%rqr&s r'r"zBltConfig.__init__>s2%'>$!2$(!1$*"4#6 0%zz*;VD!',?!F&,jj1I&J#"JJ~u=)-I,^L^)-J*4X1  !"2EV"WD  KKR S  -  % %&9;L M"2"D^"DD  (8 9"0D   !"7J["\D  KKR S  -  % %&9;L M"7"I."ID  (= >"0D   !"7J["\D  KKR S  -  % %&9;L M"7"I."ID  (= >"0D   !;N_!`D  KKP Q  t ,  $ $%8:K L!;!Lm!LD   'A B!.D %)$7$7$C$CdFWFW$W!)BdFXFXFdFd)d %jn 4  ($/ K-@KFKr))r*rITr^entropyg]?r/Nr+Ni"r/NNNNFr6r3N) r8r9r:r;r<keys_to_ignore_at_inferencerKrr@rE sub_configsr"r=r>s@r'rQrQsy@DJ#4"5*//3 K $,%)&,-.!+`L`Lr)rQ)rQrKrr@rEN)r;configuration_utilsrutilsr get_loggerr8rlrr@rErKrQ__all__r r)r'rysw3   H %/>,/>d/>,/>d'>!1'>TI>'I>XlL lL^ r)