DarthAffe/stable-diffusion.cpp
1
0 Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-11-11 11:26:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

bring back freeing pmid lora params buffer; simply pooled output of CLIPvision

Browse Source
This commit is contained in:
bssrdf committed 2024-02-27 14:49:39 -05:00
1 parent 60796158b1
commit fd098afc6d
2 files changed
+10 -10
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

clip.hpp
+4 -4
View File
@@ -904,10 +904,10 @@ public:
GGML_ASSERT(x->ne[3] == 1);
// int64_t max_token_idx = 0;
// ggml_tensor* pooled = ggml_view_1d(ctx, x, x->ne[0], x->nb[1] * max_token_idx); // assert N == 1
x = ggml_cont(ctx, ggml_permute(ctx, x, 0, 2, 1, 3));
ggml_tensor* pooled = ggml_view_2d(ctx, x, x->ne[0], x->ne[1], x->nb[0], 0);
pooled = ggml_scale(ctx, ggml_cont(ctx, pooled), 1.f);
// print_ggml_tensor(pooled, true, "pooled");
// x = ggml_cont(ctx, ggml_permute(ctx, x, 0, 2, 1, 3));
// ggml_tensor* pooled = ggml_view_2d(ctx, x, x->ne[0], x->ne[1], x->nb[0], 0);
// pooled = ggml_cont(ctx, pooled);
ggml_tensor* pooled = ggml_cont(ctx, ggml_view_2d(ctx, x, x->ne[0], x->ne[2], x->nb[2], 0));
return pooled; // [N, projection_dim]
}