pl-flow / s2 /config.json
rcell233's picture
Publish PL-Flow model bundle
6ba0577
Raw History Blame Contribute Delete
1.03 kB
{
"format_version": 1,
"type": "s2",
"model": {
"model_params": {
"style_encoder": {
"dim": 256
},
"DiT": {
"hidden_dim": 512,
"num_heads": 8,
"depth": 13,
"class_dropout_prob": 0.1,
"in_channels": 32,
"content_dim": 32
},
"wavenet": {
"hidden_dim": 512,
"num_layers": 8,
"kernel_size": 5,
"dilation_rate": 1,
"p_dropout": 0.2
}
},
"conditioner": {
"n_vocab": 178,
"output_channels": 32,
"ssl_dim": 768,
"hidden_channels": 192,
"prosody_dim": 16,
"duration_bins": 512,
"filter_channels": 768,
"n_heads": 2,
"n_layers": 6,
"kernel_size": 3,
"p_dropout": 0.1,
"prenet": true,
"duration_loss_weight": 1.0,
"max_token_duration": 511,
"max_infer_frames": 1000
},
"prompt": {
"min_seconds": 1.0,
"max_seconds": 5.0,
"max_fraction": 0.5,
"dropout": 0.1
}
}
}