Skip to content

Commit 4348323

Browse files
committed
Added Params-Extension for FreeParamsImmediately
1 parent e6297d4 commit 4348323

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

StableDiffusion.NET/Models/Parameter/Extensions/DiffusionModelBuilderExtension.cs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,13 @@ public static DiffusionModelParameter WithVaeDecodeOnly(this DiffusionModelParam
8080
return parameter;
8181
}
8282

83+
public static DiffusionModelParameter WithImmediatelyFreedParams(this DiffusionModelParameter parameter, bool immediatelyFreedParams = true)
84+
{
85+
parameter.FreeParamsImmediately = immediatelyFreedParams;
86+
87+
return parameter;
88+
}
89+
8390
public static DiffusionModelParameter WithVaeTiling(this DiffusionModelParameter parameter, bool vaeTiling = true)
8491
{
8592
parameter.VaeTiling = vaeTiling;

0 commit comments

Comments
 (0)