Function parse_config

src/models/miocodec/assets.cpp:60–108

    config.global_encoder_intermediate_dim = yaml::require_int(parsed, "model.init_args.global_encoder.init_args.intermediate_dim");

    config.wave_prenet_dim = yaml::require_int(parsed, "model.init_args.wave_prenet.init_args.dim");
    config.wave_prenet_output_dim = yaml::require_int(parsed, "model.init_args.wave_prenet.init_args.output_dim");
    config.wave_prenet_layers = yaml::require_int(parsed, "model.init_args.wave_prenet.init_args.n_layers");
    config.wave_prenet_heads = yaml::require_int(parsed, "model.init_args.wave_prenet.init_args.n_heads");
    config.wave_prenet_window_size = yaml::require_int(parsed, "model.init_args.wave_prenet.init_args.window_size");
    config.wave_prenet_max_seq_len = yaml::require_int(parsed, "model.init_args.wave_prenet.init_args.max_seq_len");

    config.wave_decoder_layers = yaml::require_int(parsed, "model.init_args.wave_decoder.init_args.n_layers");
    config.wave_decoder_heads = yaml::require_int(parsed, "model.init_args.wave_decoder.init_args.n_heads");
    config.wave_decoder_window_size = yaml::require_int(parsed, "model.init_args.wave_decoder.init_args.window_size");
    config.wave_decoder_max_seq_len = yaml::require_int(parsed, "model.init_args.wave_decoder.init_args.max_seq_len");
    config.wave_decoder_condition_dim = yaml::require_int(parsed, "model.init_args.wave_decoder.init_args.adanorm_condition_dim");
    return config;
}

}  // namespace

std::shared_ptr<const MioCodecAssets> load_miocodec_assets(const std::filesystem::path & model_path) {
    auto assets = std::make_shared<MioCodecAssets>();
    assets->resources = engine::model_spec::load_resource_bundle_for_family(model_path, "miocodec");
    assets->config = parse_config(assets->resources);
    assets->model_weights = assets->resources.open_tensor_source("weights");
    assets->wavlm_weights = assets->resources.open_tensor_source("wavlm_weights");
    return assets;
}

}  // namespace engine::models::miocodec