From 54cea07af51237eb2bde3e3a0c8c89fa1e9471aa Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Thu, 25 Jan 2024 05:16:33 -0800
Subject: [PATCH] Update forge_loader.py

---
 modules_forge/forge_loader.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py
index 71027236..7ae02258 100644
--- a/modules_forge/forge_loader.py
+++ b/modules_forge/forge_loader.py
@@ -205,6 +205,8 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
     sd_model.sd_checkpoint_info = checkpoint_info
     timer.record("forge finalize")
 
+    sd_model.unet_patcher = forge_object.unet
+
     return sd_model