Skip to content

Commit 004eaba

Browse files
committed
update doc comment with padding info for ggml_timestep_embedding
1 parent 991ab9c commit 004eaba

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/include/ggml.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2122,7 +2122,7 @@ extern "C" {
21222122

21232123
// Ref: https://github.com/CompVis/stable-diffusion/blob/main/ldm/modules/diffusionmodules/util.py#L151
21242124
// timesteps: [N,]
2125-
// return: [N, dim]
2125+
// return: [N, PAD(dim, 2)]
21262126
GGML_API struct ggml_tensor * ggml_timestep_embedding(
21272127
struct ggml_context * ctx,
21282128
struct ggml_tensor * timesteps,

0 commit comments

Comments
 (0)