o
    پii                     @   sH   d Z 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	dS )z
Generic sampling parameters for diffusers backend.

This module provides generic sampling parameters that work with any diffusers pipeline.
    )	dataclassfield)Any)DataTypeSamplingParamsc                   @   s~   e Zd ZU dZ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Zeed< eedZeeef ed< dddZdS )DiffusersGenericSamplingParamsaa  
    Generic sampling parameters for diffusers backend.

    These parameters cover the most common options across different diffusers pipelines.
    The diffusers pipeline will use whichever parameters it supports.

    For pipeline-specific parameters, use `diffusers_kwargs` dict which will be
    passed directly to the diffusers pipeline call.
       
num_frames   heightwidth   num_inference_stepsg      @guidance_scale negative_prompt)default_factorydiffusers_kwargsreturnNc                 C   sP   | j dkr
tj| _ntj| _| jd u rd| _d| _| jd u r&d| _d| _d S d S )Nr   Tr
   )	r	   r   VIDEO	data_typeIMAGEr   width_not_providedr   height_not_provided)self r   j/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/configs/sample/diffusers_generic.py__post_init__)   s   




z,DiffusersGenericSamplingParams.__post_init__)r   N)__name__
__module____qualname____doc__r	   int__annotations__r   r   r   r   floatr   strr   dictr   r   r   r   r   r   r   r      s   
 r   N)
r!   dataclassesr   r   typingr   4sglang.multimodal_gen.configs.sample.sampling_paramsr   r   r   r   r   r   r   <module>   s   