o
    §µiO  ã                   @   sà   d dl Z d dlZej e  ¡ ¡ d dlZd dlZd dlmZm	Z	 	 	 e	ddddddd	Z
ee
d
ZdZdZdZdZeee e ƒZdZejee dee¡ejdeejdfd\ZZeded › dƒ eded › dƒ dS )é    N)ÚCFMÚDiTi   é   é   é   i   é   )ÚdimÚdepthÚheadsÚff_multÚtext_dimÚconv_layers)ÚtransformeriÀ]  éd   é   é   é–   é   )Údtype)ÚinputszFLOPs: g    eÍÍAz GzParams: g    €„.Az M)ÚosÚsysÚpathÚappendÚgetcwdÚthopÚtorchÚf5_tts.modelr   r   r   ÚmodelÚtarget_sample_rateÚn_mel_channelsÚ
hop_lengthÚdurationÚintÚframe_lengthÚtext_lengthÚprofileÚrandnÚzerosÚlongÚflopsÚparamsÚprint© r-   r-   úV/home/ubuntu/.local/lib/python3.10/site-packages/f5_tts/scripts/count_params_gflops.pyÚ<module>   s*    
 
ÿ