Function bind_upsampler

src/models/miocodec/weights.cpp:461–492

}

MioCodecUpsamplerWeights bind_upsampler(
    MioCodecWeights & weights,
    const engine::assets::TensorSource & source,
    engine::assets::TensorStorageType storage_type) {
    MioCodecUpsamplerWeights upsampler;
    const int64_t factors[] = {3, 3};
    const int64_t kernels[] = {9, 9};
    int64_t in_channels = 512;
    upsampler.stages.reserve(2);
    for (int64_t stage = 0; stage < 2; ++stage) {
        const int64_t out_channels = in_channels / 2;
        const std::string stage_index = std::to_string(stage);
        MioCodecUpsamplerStageWeights item;
        item.upsample = bind_weight_norm_conv_transpose1d(
            weights,
            source,
            "wave_upsampler.upsample_layers." + stage_index,
            in_channels,
            out_channels,
            kernels[stage],
            static_cast<int>(factors[stage]),
            static_cast<int>((kernels[stage] - factors[stage]) / 2),
            storage_type);
        item.snake = bind_snake_beta(weights, source, "wave_upsampler.snake_activations." + stage_index, out_channels);
        item.resnet = bind_resnet_block(weights, "wave_upsampler.resnet_blocks." + stage_index, out_channels);
        upsampler.stages.push_back(std::move(item));
        in_channels = out_channels;
    }
    upsampler.output_projection = bind_linear(weights, "wave_upsampler.out_proj", 128, 512);
    upsampler.output_snake = bind_snake_beta(weights, source, "wave_upsampler.out_snake", 512);
    return upsampler;
}

MioCodecGlobalEncoderWeights bind_global_encoder(const MioCodecWeights & weights) {
    MioCodecGlobalEncoderWeights global;