Function load_residual_unit

src/models/ace_step/vae_decoder.cpp:308–320

    out.snake2 = load_snake_exact(store, source, prefix + ".snake2", channels);
    out.conv2 = load_weight_norm_conv1d(store, source, prefix + ".conv2", channels, channels, 1, 1, 0, 1, storage_type, true);
    return out;
}

VAEDecoderWeights load_vae_decoder_weights(
    const AceStepAssets & assets,
    ggml_backend_t backend,
    core::BackendType backend_type,
    size_t weight_context_bytes,
    assets::TensorStorageType storage_type) {
    const auto & config = assets.config.vae;
    if (config.decoder_channels <= 0 ||
        config.decoder_input_channels <= 0 ||
        config.channel_multiples.empty() || config.downsampling_ratios.empty()) {
        throw std::runtime_error("ACE-Step VAE config is incomplete for decoder loading");
    }
    const auto & source = *assets.vae_weights;