o
    ïºj4  ã                   @  sB   d Z ddlmZ ddlmZmZ ddlmZ eG dd„ dƒƒZdS )z¹V3 face model configuration.

Hyperparameters chosen to land at ~3M params (~12 MB fp32, ~3 MB int8) with
~1 ms/frame CPU inference. See architecture sketch in models/v3_face/model.py.
é    )Úannotations)Ú	dataclassÚfield)ÚTuplec                   @  s  e Zd ZU dZded< dZded< dZded< dZded	< d
Zded< dZ	ded< dZ
ded< dZded< dZded< dZded< dZded< dZded< dZded< dZded< d Zded!< d"Zded#< d$Zded%< d"Zded&< d"Zded'< d(Zd)ed*< d+Zded,< d"Zded-< d.Zded/< d0Zded1< d2Zded3< d4Zded5< d6Zded7< d8Zded9< d(Z d)ed:< d(Z!d)ed;< d.Z"ded<< d=Z#ded>< d=Z$ded?< d=Z%ded@< d=Z&dedA< d(Z'd)edB< d(Z(d)edC< d(Z)d)edD< d(Z*d)edE< d(Z+d)edF< dGZ,dHedI< d2Z-dedJ< d=Z.dedK< d(S )LÚV3FaceConfigéP   ÚintÚ	audio_dimé   Úcond_dimé4   Ú
output_diméÀ   Ú
hidden_dimé   Úkernel_size)é   é   é   é   é   é    zTuple[int, ...]Úshared_dilations)é@   é€   Úbranch_dilationsgš™™™™™¹?ÚfloatÚdropouté   Úfpséð   Úcrop_framesr   Ú
batch_sizegü©ñÒMbP?Úlearning_rateg{®Gáz„?Úweight_decayiô  Úwarmup_stepsé2   Ún_epochsg      ð?Ú	grad_clipg      @Úlong_oversample_weightÚlipsync_target_gainÚexpression_target_gainNzfloat | NoneÚemotional_mouth_target_gaing       @Úlipsync_weightÚexpression_weightg333333Ó?Úeye_blink_weighté   Úvelocity_warmup_epochsgffffffæ?Úvelocity_lipsync_weightgš™™™™™á?Úvelocity_shared_weightgÍÌÌÌÌÌÜ?Úvelocity_expression_weightgš™™™™™©?Úvelocity_eye_blink_weightÚvelocity_eye_squint_weightÚvelocity_brow_weightÚvelocity_weightg        Úplosive_damp_targetÚsmooth_target_sigma_browÚsmooth_target_sigma_eye_squintÚsmooth_target_sigma_eye_wideÚbrow_innerup_happy_gainÚbrow_happy_gainÚbrow_target_gainÚbrow_surprise_gainÚeye_wide_maxFÚboolÚ	soft_clipÚsoft_clip_kneeÚsmooth_cond_sigma_emotion)/Ú__name__Ú
__module__Ú__qualname__r	   Ú__annotations__r   r   r   r   r   r   r   r   r!   r"   r#   r$   r%   r'   r(   r)   r*   r+   r,   r-   r.   r/   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rC   rD   rE   © rJ   rJ   úM/dataset/kemix-engine/package/face/animasync-face-v3/models/v3_face/config.pyr      sX   
 	

r   N)	Ú__doc__Ú
__future__r   Údataclassesr   r   Útypingr   r   rJ   rJ   rJ   rK   Ú<module>   s    