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