o ٷid@s6ddlmZmZddlmZmZeGdddZdS)) dataclassfield)OmniDiffusionSamplingParamsOmniPromptTypec@sDeZdZUdZeeed<eed<eedZ ee ed<ddZ dS) OmniDiffusionRequesta Complete state passed through the pipeline execution. This dataclass contains the prompts and sampling parameters for the diffusion pipeline execution. It also contains a request_id for other components to trace this request and its outputs. promptssampling_params)default_factory request_idscCs`|jjdkrtdd|jDrd|j_|jjdur |jj|j_|jjr*d|j_dSd|j_dS)z;Initialize dependent fields after dataclass initialization.g?css&|]}t|t o|dVqdS)negative_promptN) isinstancestrget).0prO/home/ubuntu/.local/lib/python3.10/site-packages/vllm_omni/diffusion/request.py s z5OmniDiffusionRequest.__post_init__..TN)rguidance_scaleanyrdo_classifier_free_guidanceguidance_scale_2guidance_scale_provided)selfrrr __post_init__s    z"OmniDiffusionRequest.__post_init__N) __name__ __module__ __qualname____doc__listr__annotations__rrr r rrrrrr s   rN) dataclassesrrvllm_omni.inputs.datarrrrrrrs