
    ri                         S SK r S SKrS SKJr  SSKJr  SSKJr  \ R                  " \	5      r
 " S S\5      rS r\S	:X  a  \" 5         gg)
    N)LightningCLI   )VitsDataModule)	VitsModelc                       \ rS rSrS rSrg)VitsLightningCLI   c                 $   UR                  SS5        UR                  SS5        UR                  SS5        UR                  SS5        UR                  S	S
5        UR                  SS5        UR                  SS5        UR                  SS5        g )Nzdata.batch_sizezmodel.batch_sizezdata.num_symbolszmodel.num_symbolszmodel.num_speakerszdata.num_speakerszmodel.sample_ratezdata.sample_ratezmodel.filter_lengthzdata.filter_lengthzmodel.hop_lengthzdata.hop_lengthzmodel.win_lengthzdata.win_lengthzmodel.segment_sizezdata.segment_size)link_arguments)selfparsers     Z/home/kim/smarthome/piper_models/venv/lib/python3.13/site-packages/piper/train/__main__.pyadd_arguments_to_parser(VitsLightningCLI.add_arguments_to_parser   s    /1CD02EF24GH13EF35IJ02CD02CD24GH     N)__name__
__module____qualname____firstlineno__r   __static_attributes__r   r   r   r   r      s    Ir   r   c                  @   [         R                  " [         R                  S9  S[        R                  R
                  R                  l        S[        R                  R                  l        S[        R                  R                  l	        [        [        [        SS0S9n g )N)levelTF
max_epochs)trainer_defaults)loggingbasicConfigINFOtorchbackendscudamatmul
allow_tf32cudnndeterministicr   r   r   )_clis    r   mainr(      sf    gll+,0ENN)&*ENN#).ENN&>\24FDr   __main__)r   r    lightning.pytorch.clir   vits.datasetr   vits.lightningr   	getLogger__package___LOGGERr   r(   r   r   r   r   <module>r0      sK      . ( %


K
(	I| 	I zF r   