Skip to content

Commit 56339d4

Browse files
committed
Wan2.1 decode not crashing anymore (still broken)
1 parent bd0c5e2 commit 56339d4

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
@@ -3837,7 +3837,7 @@ SD_API sd_image_t* generate_video(sd_ctx_t* sd_ctx, const sd_vid_gen_params_t* s
38373837
struct ggml_tensor* vid = sd_ctx->sd->decode_first_stage(work_ctx, final_latent, true);
38383838
int64_t t5 = ggml_time_ms();
38393839
LOG_INFO("decode_first_stage completed, taking %.2fs", (t5 - t4) * 1.0f / 1000);
3840-
if (sd_ctx->sd->free_params_immediately) {
3840+
if (sd_ctx->sd->free_params_immediately && !sd_ctx->sd->use_tiny_autoencoder) {
38413841
sd_ctx->sd->first_stage_model->free_params_buffer();
38423842
}
38433843

0 commit comments

Comments
 (0)