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;