File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments