o پi'@s UddlmZddlmZmZddlmZddlZddlm Z m Z m Z ddl m Z ddlmZmZmZddlmZdd lmZmZd ZGd d d eZed dZeed<dedefddZdedejfddZdedefddZdedejfddZ eGdddeZ!eGddde!Z"dS))Callable) dataclassfield) TypedDictN) DiTConfig EncoderConfig VAEConfig)HunyuanVideoConfig)BaseEncoderOutputCLIPTextConfig LlamaConfig)HunyuanVAEConfig) ModelTaskTypePipelineConfiga<|start_header_id|>system<|end_header_id|> Describe the video by detailing the following aspects: 1. The main content and theme of the video.2. The color, shape, size, texture, quantity, text, and spatial relationships of the objects.3. Actions, events, behaviors temporal relationships, physical movement changes of the objects.4. background environment, light, style and atmosphere.5. camera angles, movements, and transitions used in the video:<|eot_id|><|start_header_id|>user<|end_header_id|> {}<|eot_id|>c@seZdZUeed<eed<dS)PromptTemplatetemplate crop_startN)__name__ __module__ __qualname__str__annotations__intrrj/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/configs/pipeline_configs/hunyuan.pyr"s  r_)rrprompt_template_videopromptreturncCstd|S)Nr)rformatrrrrllama_preprocess_text-sr!outputscCsJd}|jdus J|j}||d }tdd}|dd|df}|S)Nr) hidden_statesrget)r" _text_inputshidden_state_skip_layerr&last_hidden_staterrrrllama_postprocess_text1s r+cCs|SNrr rrrclip_preprocess_text;sr-cCs |j}|Sr,) pooler_output)r"r(r.rrrclip_postprocess_text?sr/c@s eZdZUdZejZeed<ee dZ e ed<ee dZ eed<dZeed<dZeed <ed d dZeed fed <edd dZeeegefd fed<edd dZeeegejfd fed<dZeed<dZeed<edd dZeed fed<ddZdS) HunyuanConfigz5Base configuration for HunYuan pipeline architecture. task_type)default_factory dit_config vae_configembedded_cfg_scale flow_shiftcCs ttfSr,)r r rrrrUs zHunyuanConfig..text_encoder_configscCttfSr,)r!r-rrrrr9Xpreprocess_text_funcscCr;r,)r+r/rrrrr9[r<postprocess_text_funcsbf16 dit_precisionfp16 vae_precisioncCsdS)N)rArArrrrrr9bstext_encoder_precisionscCsd|j_d|j_dS)NFT)r4 load_encoder load_decoder)selfrrr __post_init__es zHunyuanConfig.__post_init__N) rrr__doc__rT2Vr1rrr r3rr r4rr6rr8r:tuplerr=rrr>r torchtensorr@rBrCrGrrrrr0Ds*       r0c@seZdZUdZdZeed<dS)FastHunyuanConfigz=Configuration specifically optimized for FastHunyuan weights.r8N)rrrrHr8rrrrrrrMjs rM)#collections.abcr dataclassesrrtypingrrK$sglang.multimodal_gen.configs.modelsrrr)sglang.multimodal_gen.configs.models.ditsr -sglang.multimodal_gen.configs.models.encodersr r r )sglang.multimodal_gen.configs.models.vaesr 3sglang.multimodal_gen.configs.pipeline_configs.baserrPROMPT_TEMPLATE_ENCODE_VIDEOrrrrr!rLr+r-r/r0rMrrrrs.     %