refactor: introduce ggml_ext_zeros_like/ggml_ext_ones_like (#1312)

This commit is contained in:
leejet authored and GitHub committed 2026-03-04 00:36:52 +08:00
1 parent d41f5fff69
commit ba35dd734e
3 files changed
+13 -3

No files matched your search

+1 -1
View File
@@ -404,7 +404,7 @@ namespace Qwen {
auto t_emb = time_text_embed->forward(ctx, timestep);
if (params.zero_cond_t) {
auto t_emb_0 = time_text_embed->forward(ctx, ggml_ext_zeros(ctx->ggml_ctx, timestep->ne[0], timestep->ne[1], timestep->ne[2], timestep->ne[3]));
auto t_emb_0 = time_text_embed->forward(ctx, ggml_ext_zeros_like(ctx->ggml_ctx, timestep));
t_emb = ggml_concat(ctx->ggml_ctx, t_emb, t_emb_0, 1);
}
auto img = img_in->forward(ctx, x);