ó
    <Ð˜j*  ã                   óü   • S SK r S SK Jr  S SKJrJr  S SKJr  S SKJrJ	r	  S SK
Jr  S SKJr  S SKJr  S S	KJr  S S
KJrJr  S SKJrJr  S SKJr  S SKJr  S SKJr  S SKJr  \\\	\\\\\\\\\\\\4r  " S S\5      r!g)é    N)Únn)ÚModuleÚ
ModuleList)ÚVectorQuantize)Ú
ResidualVQÚGroupedResidualVQ)ÚRandomProjectionQuantizer)ÚFSQ)ÚFSP)ÚLFQ)ÚResidualLFQÚGroupedResidualLFQ)ÚResidualFSQÚGroupedResidualFSQ)ÚLatentQuantize)ÚSimVQ)ÚResidualSimVQ)ÚHierarchicalVQc                   ó6   ^ • \ rS rSrS\4U 4S jjrS rSrU =r$ )Ú
Sequentialé(   Úfnsc                 óÈ   >• [         TU ]  5         [        U Vs/ s H  n[        [	        U[
        5      5      PM     sn5      S:X  d   S5       e[        U5      U l        g s  snf )Né   z:this special Sequential must contain exactly one quantizer)ÚsuperÚ__init__ÚsumÚintÚ
isinstanceÚQUANTIZE_KLASSESr   r   )Úselfr   ÚfnÚ	__class__s      €Ú«C:\Users\johnn\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.13_qbz5n2kfra8p0\LocalCache\local-packages\Python313\site-packages\vector_quantize_pytorch/utils.pyr   ÚSequential.__init__)   s`   ø€ ô 	‰ÑÔÜÁCÓHÂC¸b”Cœ
 2Ô'7Ó8Ö9ÁCÑHÓIÈQÓNð  	Mð  QMó  	MÐNä˜c“?ˆ�ùò Is   ™#Ac                 óŠ   • U R                    H,  n[        U[        5      (       d
  U" U5      nM"  U" U40 UD6tpM.     U/WQ7nU$ )N)r   r   r    )r!   ÚxÚkwargsr"   ÚrestÚoutputs         r$   ÚforwardÚSequential.forward2   sO   € ð
 —(”(ˆBä˜bÔ"2×3Ñ3Ù�q“E�Ùá˜!‘˜v‘ˆHˆA‘ñ ð ��d‘ˆàˆó    )r   )	Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   r   r+   Ú__static_attributes__Ú__classcell__)r#   s   @r$   r   r   (   s   ø† ð#à÷#÷ð r-   r   )"Útorchr   Útorch.nnr   r   Ú/vector_quantize_pytorch.vector_quantize_pytorchr   Ú#vector_quantize_pytorch.residual_vqr   r   Ú3vector_quantize_pytorch.random_projection_quantizerr	   Ú2vector_quantize_pytorch.finite_scalar_quantizationr
   Ú2vector_quantize_pytorch.finite_scalar_perturbationr   Ú0vector_quantize_pytorch.lookup_free_quantizationr   Ú$vector_quantize_pytorch.residual_lfqr   r   Ú$vector_quantize_pytorch.residual_fsqr   r   Ú+vector_quantize_pytorch.latent_quantizationr   Úvector_quantize_pytorch.sim_vqr   Ú'vector_quantize_pytorch.residual_sim_vqr   Ú'vector_quantize_pytorch.hierarchical_vqr   r    r   © r-   r$   Ú<module>rC      sx   ðÛ Ý ß 'õ Kß MÝ YÝ BÝ BÝ @ß Pß PÝ FÝ 0Ý AÝ Bð ØØØØØØ	ØØØØØØØØðÐ ô(�õ r-   