Skip to content

Commit 6a653e6

Browse files
committed
fix img2vid
1 parent a7a791d commit 6a653e6

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

stable-diffusion.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3633,7 +3633,8 @@ SD_API sd_image_t* generate_video(sd_ctx_t* sd_ctx, const sd_vid_gen_params_t* s
36333633
denoise_mask = ggml_new_tensor_4d(work_ctx, GGML_TYPE_F32, init_latent->ne[0], init_latent->ne[1], init_latent->ne[2], 1);
36343634
ggml_set_f32(denoise_mask, 1.f);
36353635

3636-
sd_ctx->sd->process_latent_out(init_latent);
3636+
if (!sd_ctx->sd->use_tiny_autoencoder)
3637+
sd_ctx->sd->process_latent_out(init_latent);
36373638

36383639
ggml_ext_tensor_iter(init_image_latent, [&](ggml_tensor* t, int64_t i0, int64_t i1, int64_t i2, int64_t i3) {
36393640
float value = ggml_ext_tensor_get_f32(t, i0, i1, i2, i3);
@@ -3643,7 +3644,8 @@ SD_API sd_image_t* generate_video(sd_ctx_t* sd_ctx, const sd_vid_gen_params_t* s
36433644
}
36443645
});
36453646

3646-
sd_ctx->sd->process_latent_in(init_latent);
3647+
if (!sd_ctx->sd->use_tiny_autoencoder)
3648+
sd_ctx->sd->process_latent_in(init_latent);
36473649

36483650
int64_t t2 = ggml_time_ms();
36493651
LOG_INFO("encode_first_stage completed, taking %" PRId64 " ms", t2 - t1);

0 commit comments

Comments
 (0)