{ "activation_dropout": 0.1, "architectures": [ "VitsModel" ], "attention_dropout": 0.1, "duration_predictor_dropout": 0.5, "duration_predictor_filter_channels": 256, "duration_predictor_kernel_size": 3, "duration_predictor_num_flows": 4, "encoder_attention_heads": 2, "encoder_ffn_dim": 768, "encoder_layerdrop": 0.1, "encoder_layers": 6, "ffn_kernel_size": 3, "flow_size": 192, "hidden_act": "relu", "hidden_dropout": 0.1, "hidden_size": 192, "initializer_range": 0.02, "layer_norm_eps": 1e-05, "leaky_relu_slope": 0.1, "model_type": "vits", "num_speakers": 1, "posterior_encoder_num_wavenet_layers": 16, "prior_encoder_num_flows": 4, "prior_encoder_num_wavenet_layers": 4, "resblock_dilation_sizes": [ [ 1, 3, 5 ], [ 1, 3, 5 ], [ 1, 3, 5 ] ], "resblock_kernel_sizes": [ 3, 7, 11 ], "speaker_embedding_size": 0, "spectrogram_bins": 513, "torch_dtype": "float32", "transformers_version": "4.31.0.dev0", "upsample_initial_channel": 512, "upsample_kernel_sizes": [ 16, 16, 4, 4 ], "upsample_rates": [ 8, 8, 2, 2 ], "use_stochastic_duration_prediction": true, "vocab_size": 178, "wavenet_dilation_rate": 1, "wavenet_dropout": 0.0, "wavenet_kernel_size": 5 }