Function expand_batch_token

src/models/heartmula/codec.cpp:709–717

    return modules::ConcatModule({1}).build(ctx, cos_part, sin_part);
}

core::TensorValue adaln_embedding(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & timesteps,
    const core::TensorValue & freqs,
    const HeartCodecAdaLayerNormWeights & weights,
    int64_t hidden_size) {
    auto embedded = timestep_embedding(ctx, timesteps, freqs);
    embedded = modules::LinearModule({512, hidden_size, true, GGML_PREC_F32}).build(ctx, embedded, weights.timestep_linear_1);
    embedded = modules::SiluModule{}.build(ctx, embedded);
    return modules::LinearModule({hidden_size, hidden_size, true, GGML_PREC_F32})
        .build(ctx, embedded, weights.timestep_linear_2);