o
    پin%                     @   s  d dl mZmZ d dlmZ d dlmZ eG dd deZeG dd deZeG dd	 d	eZ	eG d
d deZ
eG dd deZeG dd deZeG dd deZeG dd deZeG dd deZeG dd deZeG dd deZeG dd deZdS )    )	dataclassfield)SamplingParamsWanTeaCacheParamsc                   @      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< d	Z	e
ed
< dZeed< dZeed< edd dZeeeef  dB ed< edd dZeed< dS )WanT2V_1_3B_SamplingParams  height@  widthQ   
num_frames   fps      @guidance_scale  Bright tones, overexposed, static, blurred details, subtitles, style, works, paintings, images, static, overall gray, worst quality, low quality, JPEG compression residue, ugly, incomplete, extra fingers, poorly drawn hands, poorly drawn faces, deformed, disfigured, misshapen limbs, fused fingers, still picture, messy background, three legs, many people in the background, walking backwardsnegative_prompt2   num_inference_stepsc                   C      ddgS Nr   r	   r	   r    r   r   r   \/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/configs/sample/wan.py<lambda>      z#WanT2V_1_3B_SamplingParams.<lambda>default_factoryNsupported_resolutionsc                   C      t dg dg ddS )Ng{Gz?)gAcG{gAC4@g,9$5gfe+@ge)ga@gtk|g~l*i@gMܘ g{N?teacache_threshret_steps_coeffsnon_ret_steps_coeffsr   r   r   r   r   r   "   
    teacache_params__name__
__module____qualname__r
   int__annotations__r   r   r   r   floatr   strr   r   r!   listtupler(   r   r   r   r   r   r   
   s   
 
r   c                   @   r   )WanT2V_14B_SamplingParams  r
      r   r   r   r   r         @r   r   r   r   r   c                   C      g dS N))r5   r4   )r4   r5   r   r   r   r   r   r   r   r   G       z"WanT2V_14B_SamplingParams.<lambda>r   Nr!   c                   C   s   t ddg dg ddS )Ng?FgfffZg$(~@g=\rg_BF^M@gn=2Կg'>gsUI@g]Lg{:Yp@gX^E*)r$   use_ret_stepsr%   r&   r   r   r   r   r   r   P   s    r(   r)   r   r   r   r   r3   6   s   
 
	r3   c                   @   b   e Zd ZU dZeed< dZeed< edd dZ	e
eeef  dB ed	< ed
d dZeed< dS )WanI2V_14B_480P_SamplingParamr6   r   r   r   c                   C   r   r   r   r   r   r   r   r   n   r   z&WanI2V_14B_480P_SamplingParam.<lambda>r   Nr!   c                   C   r"   )Ngp=
ף?r:   r;   r#   r   r   r   r   r   r   u   r'   r(   r*   r+   r,   r   r/   r.   r   r-   r   r!   r1   r2   r(   r   r   r   r   r   r>   e   s   
 r>   c                   @   r=   )WanI2V_14B_720P_SamplingParamr6   r   r   r   c                   C   r7   r8   r   r   r   r   r   r      r9   z&WanI2V_14B_720P_SamplingParam.<lambda>r   Nr!   c                   C   r"   )Ng333333?r:   r;   r#   r   r   r   r   r   r      r'   r(   r?   r   r   r   r   r@      s   
 	r@   c                   @   sJ   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< d	Z	eed
< dS )FastWanT2V480PConfig   r   =   r   i  r
   r   r   r   r   N)
r*   r+   r,   r   r-   r.   r   r
   r   r   r   r   r   r   rA      s   
 rA   c                   @   sj   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dB ed< dZeed< dZeed< dS )"Wan2_1_Fun_1_3B_InP_SamplingParamsz2Sampling parameters for Wan2.1 Fun 1.3B InP model.r	   r
   r   r   r   r   r   r     色调艳丽，过曝，静态，细节模糊不清，字幕，风格，作品，画作，画面，静止，整体发灰，最差质量，低质量，JPEG压缩残留，丑陋的，残缺的，多余的手指，画得不好的手部，画得不好的脸部，畸形的，毁容的，形态畸形的肢体，手指融合，静止不动的画面，杂乱的背景，三条腿，背景人很多，倒着走Nr   g      @r   r   r   )r*   r+   r,   __doc__r
   r-   r.   r   r   r   r   r0   r   r/   r   r   r   r   r   rD      s   
 rD   c                   @   s"   e Zd ZU dZdZedB ed< dS )Wan2_2_Base_SamplingParams-Sampling parameters for Wan2.2 TI2V 5B model.rE   Nr   )r*   r+   r,   rF   r   r0   r.   r   r   r   r   rG      s   
 rG   c                   @   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dd dZeeeef  dB ed< dS )Wan2_2_TI2V_5B_SamplingParamrH     r
   r5   r   y   r      r   r6   r   r   r   c                   C   r   )N)r5   rJ   )rJ   r5   r   r   r   r   r   r      r   z%Wan2_2_TI2V_5B_SamplingParam.<lambda>r   Nr!   )r*   r+   r,   rF   r
   r-   r.   r   r   r   r   r/   r   r   r!   r1   r2   r   r   r   r   rI      s   
 "rI   c                   @   sd   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ef  dB ed< dS )Wan2_2_T2V_A14B_SamplingParamg      @r   r   guidance_scale_2(   r   r   r   c                   C   r7   r8   r   r   r   r   r   r      r9   z&Wan2_2_T2V_A14B_SamplingParam.<lambda>r   Nr!   r*   r+   r,   r   r/   r.   rN   r   r-   r   r   r!   r1   r2   r   r   r   r   rM         
 "rM   c                   @   sd   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ef  dB ed< dS )Wan2_2_I2V_A14B_SamplingParam      @r   rN   rO   r   r   r   c                   C   r7   r8   r   r   r   r   r   r     r9   z&Wan2_2_I2V_A14B_SamplingParam.<lambda>r   Nr!   rP   r   r   r   r   rR      rQ   rR   c                   @   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< dS )	#Turbo_Wan2_2_I2V_A14B_SamplingParamrS   r   rN      r   r   r   N)
r*   r+   r,   r   r/   r.   rN   r   r-   r   r   r   r   r   rT     s
   
 rT   c                   @   s   e Zd ZdS )SelfForcingWanT2V480PConfigN)r*   r+   r,   r   r   r   r   rV     s    rV   N)dataclassesr   r   4sglang.multimodal_gen.configs.sample.sampling_paramsr   -sglang.multimodal_gen.configs.sample.teacacher   r   r3   r>   r@   rA   rD   rG   rI   rM   rR   rT   rV   r   r   r   r   <module>   s6   +.#%
