Added Params-Extension for FreeParamsImmediately

This commit is contained in:
DarthAffe committed 2025-10-27 19:56:39 +01:00
1 parent e6297d485f
commit 4348323abc
1 file changed
+7
@@ -80,6 +80,13 @@ public static class DiffusionModelBuilderExtension
return parameter; 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) public static DiffusionModelParameter WithVaeTiling(this DiffusionModelParameter parameter, bool vaeTiling = true)
{ {
parameter.VaeTiling = vaeTiling; parameter.VaeTiling = vaeTiling;