resolve #926

backend/nn/unet.py

@@ -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:

ldm_patched/contrib/external_freelunch.py

@@ -78,7 +78,7 @@ class FreeU_V2:
     CATEGORY = "model_patches"
 
     def patch(self, model, b1, b2, s1, s2):
-        model_channels = model.model.model_config.unet_config["model_channels"]
+        model_channels = model.model.diffusion_model.legacy_config["model_channels"]
         scale_dict = {model_channels * 4: (b1, s1), model_channels * 2: (b2, s2)}
         on_cpu_devices = {}