Skip to content

Commit 6795acc

Browse files
committed
Wan2.1 decode not crashing anymore (still broken)
1 parent ea34cdb commit 6795acc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

stable-diffusion.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3125,7 +3125,7 @@ SD_API sd_image_t* generate_video(sd_ctx_t* sd_ctx, const sd_vid_gen_params_t* s
31253125
struct ggml_tensor* vid = sd_ctx->sd->decode_first_stage(work_ctx, final_latent, true);
31263126
int64_t t5 = ggml_time_ms();
31273127
LOG_INFO("decode_first_stage completed, taking %.2fs", (t5 - t4) * 1.0f / 1000);
3128-
if (sd_ctx->sd->free_params_immediately) {
3128+
if (sd_ctx->sd->free_params_immediately && !sd_ctx->sd->use_tiny_autoencoder) {
31293129
sd_ctx->sd->first_stage_model->free_params_buffer();
31303130
}
31313131

0 commit comments

Comments
 (0)