+
    œ°³jW  ã                   óx   € ^ RI Ht ^RIHt ^RIHt ^RIHtHt ]! RR7      ] ! R R]4      4       4       t	R.t
R	# )
é    )Ústrict)ÚPreTrainedConfig)Úauto_docstring)ÚCONFIG_MAPPINGÚ
AutoConfigzlerobot/pi0_base)Ú
checkpointc                   óŒ   a a€ ] tR t^t oRtRtR]R]/tRtRt	^2t
^ t^ t^
tRtRtRtR	tR
tRtRtV 3R ltR tV3R ltRtVtV ;t# )Ú	PI0ConfigaX  
vlm_config (`dict`, *optional*):
    Configuration for the vlm backbone (PaliGemmaModel).
dit_config (`dict`, *optional*):
    Configuration for the DiT backbone. Defaults to a Gemma 300M variant.
chunk_size (`int`, *optional*, defaults to 50):
    Number of action steps to predict per chunk.
max_state_dim (`int`, *optional*, defaults to 32):
    Maximum state vector dimension (shorter vectors are zero-padded).
max_action_dim (`int`, *optional*, defaults to 32):
    Maximum action vector dimension (shorter vectors are zero-padded).
num_inference_steps (`int`, *optional*, defaults to 10):
    Number of denoising steps during inference.
time_sampling_beta_alpha (`float`, *optional*, defaults to 1.5):
    Alpha parameter for Beta distribution used to sample diffusion time during training.
time_sampling_beta_beta (`float`, *optional*, defaults to 1.0):
    Beta parameter for Beta distribution used to sample diffusion time during training.
time_sampling_scale (`float`, *optional*, defaults to 0.999):
    Scale factor for sampled time values.
time_sampling_offset (`float`, *optional*, defaults to 0.001):
    Offset added to sampled time values.
min_period (`float`, *optional*, defaults to 0.004):
    Minimum period for sinusoidal time embedding.
max_period (`float`, *optional*, defaults to 4.0):
    Maximum period for sinusoidal time embedding.
loss_reduction (`str`, *optional*, defaults to `"mean"`):
    The reduction to use on MSE loss.

Example:
```python
>>> from transformers import PI0ForConditionalGeneration, PI0Config

>>> config = PI0Config()
>>> model = PI0ForConditionalGeneration(config)
```
Úpi0Ú
vlm_configÚ
dit_configNg      ø?g      ð?g+‡ÙÎ÷ï?gü©ñÒMbP?gü©ñÒMbp?g      @Úmeanc                óH  <€ \        V P                  \        4      '       d@   V P                  P                  R R4      p\        V,          ! R/ V P                  B V n        MJV P                  f=   \        R,          ! R RRRR^RRR	^R
^RR/R RRRRRR^R^àR^R	^RRRR/	RRR7      V n        \        V P
                  \        4      '       d@   V P
                  P                  R R4      p\        V,          ! R/ V P
                  B V n        MKV P
                  f>   \        R,          ! R^R^^RV P                  P                  P                  R7      V n        RV P
                  n        RV P
                  n	        RV P                  P                  n	        \        SV `,  ! R/ VB  R# )Ú
model_typeÚ	paligemmaNÚgemmaÚhidden_sizei   Únum_hidden_layersÚintermediate_sizei @  Únum_attention_headsÚnum_key_value_headsÚ
vocab_sizei€ì Úsiglip_vision_modeliÐ  i€  Ú
patch_sizeÚ
image_sizeÚvision_use_headF)Útext_configÚvision_configÚprojection_dimÚimage_token_idi   i   é   )r   r   r   r   r   Úhead_dimr   T© )Ú
isinstancer   ÚdictÚgetr   r   r   r   Ú	is_causalÚuse_bidirectional_attentionÚsuperÚ__post_init__)ÚselfÚkwargsÚvlm_model_typeÚdit_model_typeÚ	__class__s   &,  €Új/home/luca/screen-checking/.venv/lib/python3.14/site-packages/transformers/models/pi0/configuration_pi0.pyr*   ÚPI0Config.__post_init__T   s  ø€ Ü�d—o‘o¤t×,Ò,Ø!Ÿ_™_×0Ñ0°¸{ÓKˆNÜ,¨^Ö<ÑO¸t¿¹ÑOˆD�OØ�_‰_Ò$Ü,¨[Ö9à  'Ø! 4Ø'¨Ø'¨Ø)¨1Ø)¨1Ø  &ðð !Ð"7Ø'¨Ø! 4Ø  "Ø  #Ø'¨Ø)¨2Ø  &Ø% uð
ð  $Ø%ô-ˆDŒOô2 �d—o‘o¤t×,Ò,Ø!Ÿ_™_×0Ñ0°¸wÓGˆNÜ,¨^Ö<ÑO¸t¿¹ÑOˆD�OØ�_‰_Ò$Ü,¨WÖ5Ø Ø"$Ø"&Ø$%Ø$%ØØŸ?™?×6Ñ6×AÑAôˆDŒOð %)ˆ�‰Ô!Ø6:ˆ�‰Ô3ØBFˆ�‰×#Ñ#Ô?Ü‰ÒÑ' Ô'ó    c                ó¤   € V P                   P                  ^,          ^ 8w  d.   \        RV P                  P                   P                   R24      hR# )zOPart of `@strict`-powered validation. Validates the architecture of the config.zDiT hidden dim=(z) must be divisible by 2N)r   r   Ú
ValueErrorÚconfig)r+   s   &r0   Úvalidate_architectureÚPI0Config.validate_architecture†   sE   € à�?‰?×&Ñ&¨Õ*¨aÔ/ÜÐ/°·±×0FÑ0F×0RÑ0RÐ/SÐSkÐlÓmÐmñ 0r2   c                óò   <€ V ^8„  d   Qh/ S[ S[,          R,          ;R&   S[ S[,          R,          ;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R	&   S[;R
&   S[;R&   S[;R&   S[;R&   S[;R&   # )é   Nr   r   Ú
chunk_sizeÚmax_state_dimÚmax_action_dimÚnum_inference_stepsÚtime_sampling_beta_alphaÚtime_sampling_beta_betaÚtime_sampling_scaleÚtime_sampling_offsetÚ
min_periodÚ
max_periodÚloss_reduction)r%   r   ÚintÚfloatÚstr)ÚformatÚ__classdict__s   "€r0   Ú__annotate__ÚPI0Config.__annotate__   sÞ   ø‡ ‚ ñV Ñ'Õ'¨$Õ.Ñ5ñW ñX Ñ'Õ'¨$Õ.Ñ5ñY ñZ Ññ[ ñ\ Ññ] ñ^ Ññ_ ñ` Ñ!ña ñb $Ñ)ñc ñd #Ñ(ñe ñf Ñ&ñg ñh  Ñ'ñi ñj Ññk ñl Ññm ñn Ñ òo r2   )r   r   )Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__r   r   Úsub_configsr   r   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   r*   r6   Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__Ú__classcell__)r/   rI   s   @@r0   r
   r
      sz   ù‡ € ñ#ðJ €JØ ¨\¸:ÐF€Kà15€JØ15€JØ€JØ€MØ€NØ!ÐØ&)ÐØ%(ÐØ!&ÐØ"'ÐØ€JØ€JØ €Nõ0(òdn÷W … r2   r
   N)Úhuggingface_hub.dataclassesr   Úconfiguration_utilsr   Úutilsr   Úautor   r   r
   Ú__all__r#   r2   r0   Ú<module>r[      sM   ðõ( /å 3Ý #ß -ñ Ð-Ô.ØôlnÐ ó lnó ó /ðlnð^ ˆ-‚r2   