o
    پi                     @   sX   d dl mZmZ d dlmZ d dlmZmZ eG dd deZeG dd deZ	dS )	    )	dataclassfield)List)
ArchConfigModelConfigc                   @   s   e Zd ZU dZeed< dZeed< dZeed< dZ	eed< e
d	d
 dZee ed< dZeed< e
dd
 dZee ed< dZeed< dZeed< dZeed< dZeed< dZeed< dS )DacVAEArchConfig   codebook_dimi   codebook_sizeT
continuousi   decoder_dimc                   C      g dS )N)r                r   r   r   a/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/configs/models/vaes/dac.py<lambda>       zDacVAEArchConfig.<lambda>default_factorydecoder_rates   encoder_dimc                   C   r   )N)r   r   r   r   r   r   r   r   r   r   r      r   encoder_ratesi   
hop_length
latent_dim	   n_codebooksFquantizer_dropouti  sample_rateN)__name__
__module____qualname__r	   int__annotations__r
   r   boolr   r   r   r   r   r   r   r   r   r    r!   r   r   r   r   r   
   s   
 r   c                   @   s8   e Zd ZU eedZeed< dZeed< dZ	eed< dS )DacVAEConfigr   arch_configTload_encoderload_decoderN)
r"   r#   r$   r   r   r)   r&   r*   r'   r+   r   r   r   r   r(      s   
 r(   N)
dataclassesr   r   typingr   )sglang.multimodal_gen.configs.models.baser   r   r   r(   r   r   r   r   <module>   s   