Method build

src/models/ace_step/vae_encoder.cpp:242–295

}

void AceStepVAEEncodeGraph::build(size_t graph_arena_bytes) {
    const auto & config = assets_->config.vae;
    ggml_init_params params{graph_arena_bytes, nullptr, true};
    ctx_.reset(ggml_init(params));
    if (ctx_ == nullptr) {
        throw std::runtime_error("ACE-Step VAE encoder ggml context initialization failed");
    }
    core::ModuleBuildContext ctx{ctx_.get(), "ace_step.vae_encoder", backend_type_};

    input_value_ = core::make_tensor(
        ctx,
        GGML_TYPE_F32,
        core::TensorShape::from_dims({1, config.audio_channels, audio_frames_}));
    ggml_set_input(input_value_.tensor);

    auto hidden = build_conv1d(
        ctx,
        input_value_,
        weights_->encoder_conv1,
        config.audio_channels,
        config.decoder_channels,
        true);
    for (size_t i = 0; i < weights_->encoder_blocks.size(); ++i) {
        const auto & block = weights_->encoder_blocks[i];
        const int64_t in_channels = hidden.shape.dims[1];
        const int64_t out_channels = block.conv.conv.weight.shape.dims[0];
        hidden = build_encoder_block(ctx, hidden, block, in_channels, out_channels);
    }
    hidden = build_snake1d_exact_bct(
        ctx,
        hidden,
        weights_->encoder_snake_out,
        hidden.shape.dims[1]);
    hidden = build_conv1d(
        ctx,
        hidden,
        weights_->encoder_conv2,
        hidden.shape.dims[1],
        config.encoder_hidden_size,
        true);
    output_value_ = hidden;
    latent_frames_ = output_value_.shape.dims[2];
    if (output_value_.shape.dims[1] % 2 != 0) {
        throw std::runtime_error("ACE-Step VAE encoder output channels must split into mean/scale pairs");
    }
    latent_channels_ = output_value_.shape.dims[1] / 2;
    ggml_set_output(output_value_.tensor);
    graph_ = ggml_new_graph_custom(ctx_.get(), 131072, false);
    ggml_build_forward_expand(graph_, output_value_.tensor);
    gallocr_ = ggml_gallocr_new(ggml_backend_get_default_buffer_type(backend_));
    if (gallocr_ == nullptr || !ggml_gallocr_alloc_graph(gallocr_, graph_)) {
        throw std::runtime_error("ACE-Step VAE encoder backend buffer allocation failed");
    }
}

AceStepVAEEncoderRuntimeCore::AceStepVAEEncoderRuntimeCore(
    std::shared_ptr<const AceStepAssets> assets,