diff --git a/backend/patcher/controlnet.py b/backend/patcher/controlnet.py
index b1ef2993..15ecf962 100644
--- a/backend/patcher/controlnet.py
+++ b/backend/patcher/controlnet.py
@@ -350,7 +350,7 @@ class ControlLora(ControlNet):
 
     def pre_run(self, model, percent_to_timestep_function):
         super().pre_run(model, percent_to_timestep_function)
-        controlnet_config = model.diffusion_model.legacy_config.copy()
+        controlnet_config = model.diffusion_model.config.copy()
         controlnet_config.pop("out_channels")
         controlnet_config["hint_channels"] = self.control_weights["input_hint_block.0.weight"].shape[1]
         controlnet_config["dtype"] = dtype = model.storage_dtype
diff --git a/backend/patcher/lora.py b/backend/patcher/lora.py
index 58e0e62f..cc75f5cf 100644
--- a/backend/patcher/lora.py
+++ b/backend/patcher/lora.py
@@ -246,7 +246,7 @@ def model_lora_keys_unet(model, key_map={}):
             key_map["lora_unet_{}".format(key_lora)] = k
             key_map["lora_prior_unet_{}".format(key_lora)] = k
 
-    diffusers_keys = unet_to_diffusers(model.diffusion_model.legacy_config)
+    diffusers_keys = unet_to_diffusers(model.diffusion_model.config)
     for k in diffusers_keys:
         if k.endswith(".weight"):
             unet_key = "diffusion_model.{}".format(diffusers_keys[k])
diff --git a/extensions-builtin/sd_forge_freeu/scripts/forge_freeu.py b/extensions-builtin/sd_forge_freeu/scripts/forge_freeu.py
index 6240563e..360a36a4 100644
--- a/extensions-builtin/sd_forge_freeu/scripts/forge_freeu.py
+++ b/extensions-builtin/sd_forge_freeu/scripts/forge_freeu.py
@@ -52,7 +52,7 @@ class FreeU:
 
 class FreeU_V2:
     def patch(self, model, b1, b2, s1, s2):
-        model_channels = model.model.diffusion_model.legacy_config["model_channels"]
+        model_channels = model.model.diffusion_model.config["model_channels"]
         scale_dict = {model_channels * 4: (b1, s1), model_channels * 2: (b2, s2)}
         on_cpu_devices = {}