free pmid model params when required

This commit is contained in:
bssrdf
2024-02-25 23:22:45 -05:00
parent b556d4411c
commit 4ac3df145e

View File

@@ -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());