Skip to content

Commit 4078a64

Browse files
committed
Wan2.1 decode not crashing anymore (still broken)
1 parent f6510b2 commit 4078a64

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
@@ -3871,7 +3871,7 @@ SD_API sd_image_t* generate_video(sd_ctx_t* sd_ctx, const sd_vid_gen_params_t* s
38713871
struct ggml_tensor* vid = sd_ctx->sd->decode_first_stage(work_ctx, final_latent, true);
38723872
int64_t t5 = ggml_time_ms();
38733873
LOG_INFO("decode_first_stage completed, taking %.2fs", (t5 - t4) * 1.0f / 1000);
3874-
if (sd_ctx->sd->free_params_immediately) {
3874+
if (sd_ctx->sd->free_params_immediately && !sd_ctx->sd->use_tiny_autoencoder) {
38753875
sd_ctx->sd->first_stage_model->free_params_buffer();
38763876
}
38773877

0 commit comments

Comments
 (0)