From 4348323abc47b8f28200e634291e75d031a8b34f Mon Sep 17 00:00:00 2001
From: DarthAffe <darthaffe@wyrez.org>
Date: Mon, 27 Oct 2025 19:56:39 +0100
Subject: [PATCH] Added Params-Extension for FreeParamsImmediately

---
 .../Parameter/Extensions/DiffusionModelBuilderExtension.cs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/StableDiffusion.NET/Models/Parameter/Extensions/DiffusionModelBuilderExtension.cs b/StableDiffusion.NET/Models/Parameter/Extensions/DiffusionModelBuilderExtension.cs
index 3a33c60..6f9ec7f 100644
--- a/StableDiffusion.NET/Models/Parameter/Extensions/DiffusionModelBuilderExtension.cs
+++ b/StableDiffusion.NET/Models/Parameter/Extensions/DiffusionModelBuilderExtension.cs
@@ -80,6 +80,13 @@ public static class DiffusionModelBuilderExtension
         return parameter;
     }
 
+    public static DiffusionModelParameter WithImmediatelyFreedParams(this DiffusionModelParameter parameter, bool immediatelyFreedParams = true)
+    {
+        parameter.FreeParamsImmediately = immediatelyFreedParams;
+
+        return parameter;
+    }
+
     public static DiffusionModelParameter WithVaeTiling(this DiffusionModelParameter parameter, bool vaeTiling = true)
     {
         parameter.VaeTiling = vaeTiling;