Function load_snake_exact

src/models/ace_step/vae_decoder.cpp:223–240

        core::TensorShape::from_dims({1, channels, 1}),
        assets::TensorStorageType::F32,
        snake_exp_transform(alpha));
    out.beta_inv = store.make_from_f32(
        core::TensorShape::from_dims({1, channels, 1}),
        assets::TensorStorageType::F32,
        snake_inverse_beta_transform(beta));
    return out;
}

WeightNormConv1dWeights load_weight_norm_conv1d(
    core::BackendWeightStore & store,
    const assets::TensorSource & source,
    const std::string & prefix,
    int64_t out_channels,
    int64_t in_channels,
    int64_t kernel,
    int stride,
    int padding,
    int dilation,
    assets::TensorStorageType storage_type,
    bool use_bias) {
    const auto g = source.require_f32(prefix + ".weight_g", {out_channels, 1, 1});