From 4ac3df145e9c4fd70429cb557649323ad13351dd Mon Sep 17 00:00:00 2001 From: bssrdf Date: Sun, 25 Feb 2024 23:22:45 -0500 Subject: [PATCH] free pmid model params when required --- stable-diffusion.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp index 4dafe8d9..05f16b6f 100644 --- a/stable-diffusion.cpp +++ b/stable-diffusion.cpp @@ -1701,9 +1701,9 @@ sd_image_t* txt2img(sd_ctx_t* sd_ctx, prompts_embeds = sd_ctx->sd->id_encoder(work_ctx, init_img, prompts_embeds, class_tokens_mask); LOG_INFO("id_encoder"); - // if (sd_ctx->sd->free_params_immediately) { - // sd_ctx->sd->pmid_model.free_params_buffer(); - // } + if (sd_ctx->sd->free_params_immediately) { + sd_ctx->sd->pmid_model->free_params_buffer(); + } // Encode input prompt without the trigger word for delayed conditioning prompt_text_only = sd_ctx->sd->remove_trigger_from_prompt(work_ctx, prompt); // printf("%s || %s \n", prompt.c_str(), prompt_text_only.c_str());