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

fix: keep MiniMax-H3 VAE weights resident across temporal chunks (#2103)

Browse Source
This commit is contained in:
losewayy authored and GitHub committed 2026-10-06 19:40:48 +08:00
1 parent 1da55ebde8
commit 0963357214
2 files changed
+35 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/model/vae/vae.hpp
+26 -2
View File
@@ -241,6 +241,15 @@ public:
sd_tiling_params_t tiling_params,
bool circular_x = false,
bool circular_y = false) {
return encode(n_threads, x, tiling_params, circular_x, circular_y, true);
}
sd::Tensor<float> encode(int n_threads,
const sd::Tensor<float>& x,
sd_tiling_params_t tiling_params,
bool circular_x,
bool circular_y,
bool auto_end) {
int64_t t0 = ggml_time_ms();
tiling_params = resolve_tiling_params(tiling_params);
sd::Tensor<float> input = x;
@@ -280,7 +289,9 @@ public:
tiling_params);
}
runner_end();
if (auto_end) {
runner_end();
}
if (output.empty()) {
LOG_ERROR("vae encode compute failed");
@@ -298,6 +309,17 @@ public:
bool circular_x = false,
bool circular_y = false,
bool silent = false) {
return decode(n_threads, x, tiling_params, decode_video, circular_x, circular_y, silent, true);
}
sd::Tensor<float> decode(int n_threads,
const sd::Tensor<float>& x,
sd_tiling_params_t tiling_params,
bool decode_video,
bool circular_x,
bool circular_y,
bool silent,
bool auto_end) {
int64_t t0 = ggml_time_ms();
tiling_params = resolve_tiling_params(tiling_params);
sd::Tensor<float> input = x;
@@ -338,7 +360,9 @@ public:
tiling_params);
}
runner_end();
if (auto_end) {
runner_end();
}
if (output.empty()) {
LOG_ERROR("vae decode compute failed");