Learn R Programming

ggmlR (version 0.6.1)

ggml_timestep_embedding: Timestep Embedding (Graph Operation)

Description

Creates sinusoidal timestep embeddings as used in diffusion models. Reference: CompVis/stable-diffusion util.py timestep_embedding

Usage

ggml_timestep_embedding(ctx, timesteps, dim, max_period = 10000L)

Value

Tensor of shape [N, dim] with timestep embeddings

Arguments

ctx

GGML context

timesteps

Input tensor of timestep values [N]

dim

Embedding dimension

max_period

Maximum period for sinusoidal embedding (default 10000)