This commit is contained in:
layerdiffusion
2024-08-02 13:55:11 -07:00
parent d1b8a2676d
commit 91543adb90
2 changed files with 2 additions and 1 deletions

View File

@@ -660,6 +660,7 @@ class IntegratedUNet2DConditionModel(nn.Module, ConfigMixin):
transformer_depth=transformer_depth,
transformer_depth_output=transformer_depth_output,
transformer_depth_middle=transformer_depth_middle,
model_channels=model_channels,
)
if context_dim is not None: