Skip to content

Commit f80f61a

Browse files
committed
fix tae-preview-only (bad merge issue)
1 parent 07c61f1 commit f80f61a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

stable-diffusion.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -484,7 +484,7 @@ class StableDiffusionGGML {
484484
vae_decode_only = false;
485485
}
486486

487-
if (high_noise_diffusion_model || sd_ctx_params->tae_preview_only) {
487+
if (high_noise_diffusion_model) {
488488
high_noise_diffusion_model->alloc_params_buffer();
489489
high_noise_diffusion_model->get_param_tensors(tensors);
490490
}
@@ -508,7 +508,7 @@ class StableDiffusionGGML {
508508
} else if (version == VERSION_CHROMA_RADIANCE) {
509509
first_stage_model = std::make_shared<FakeVAE>(vae_backend,
510510
offload_params_to_cpu);
511-
} else if (!use_tiny_autoencoder) {
511+
} else if (!use_tiny_autoencoder || sd_ctx_params->tae_preview_only) {
512512
first_stage_model = std::make_shared<AutoEncoderKL>(vae_backend,
513513
offload_params_to_cpu,
514514
model_loader.tensor_storages_types,

0 commit comments

Comments
 (0)