o
    ÔÙ¾i.  ã                   @   s.   d dl mZ d dlmZ eG dd„ dƒƒZdS )é    )Ú	dataclass)ÚOptionalc                   @   sf   e Zd ZU dZdZee ed< dZee ed< dZ	ee ed< dZ
ee ed< dZeed< d	d
„ ZdS )ÚModelOptConfiga%  Configuration for NVIDIA ModelOpt quantization operations.

    This configuration class holds parameters for ModelOpt quantization,
    checkpoint management, and model export operations.

    Args:
        quant: Quantization method/type (e.g., "fp8", "fp4")
        checkpoint_restore_path: Path to restore ModelOpt checkpoint from
        checkpoint_save_path: Path to save ModelOpt checkpoint to
        export_path: Path to export quantized model in HuggingFace format
        quantize_and_serve: Whether to quantize and serve in one step
    NÚquantÚcheckpoint_restore_pathÚcheckpoint_save_pathÚexport_pathFÚquantize_and_servec                 C   s   dS )z,Validate configuration after initialization.N© )Úselfr
   r
   úV/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/configs/modelopt_config.pyÚ__post_init__   s   zModelOptConfig.__post_init__)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   ÚstrÚ__annotations__r   r   r   r	   Úboolr   r
   r
   r
   r   r      s   
 r   N)Údataclassesr   Útypingr   r   r
   r
   r
   r   Ú<module>   s   