
    *j!                    \    d Z ddlmZ ddlmZmZ ddlmZ e G d d                      ZdS )zV3 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                     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(S )BV3FaceConfigP   int	audio_dim   cond_dim4   
output_dim   
hidden_dim   kernel_size)                   zTuple[int, ...]shared_dilations)@      branch_dilationsg?floatdropout   fps   crop_framesr   
batch_sizegMbP?learning_rateg{Gz?weight_decayi  warmup_steps2   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brow_innerup_happy_gain)'__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/   r0   r1   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>        M/dataset/kemix-engine/package/face/animasync-face-v3/models/v3_face/config.pyr   r      s         IHJ JK(<<<<<(11111G CMMMMKJMLLHI %('''' "%$$$$$''''' 154444  N"""""!!!!!& #%$$$$ &)(((($((((((,,,,,'+++++ 043333 *.---- O     "%$$$$ '*)))),/"//// -1000000rD   r   N)	__doc__
__future__r   dataclassesr   r   typingr   r   rC   rD   rE   <module>rJ      s    
 # " " " " " ( ( ( ( ( ( ( (       E1 E1 E1 E1 E1 E1 E1 E1 E1 E1rD   