From 1f4fc9ea0ccceba2e86668a22d86e63b3d262b83 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Thu, 4 Apr 2024 23:01:02 -0400
Subject: [PATCH] Fix issue with get_model_object on patched model.
---
comfy/model_patcher.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py
index 97fabd4f..1595d4a2 100644
--- a/comfy/model_patcher.py
+++ b/comfy/model_patcher.py
@@ -154,7 +154,10 @@ class ModelPatcher:
if name in self.object_patches:
return self.object_patches[name]
else:
- return comfy.utils.get_attr(self.model, name)
+ if name in self.object_patches_backup:
+ return self.object_patches_backup[name]
+ else:
+ return comfy.utils.get_attr(self.model, name)
def model_patches_to(self, device):
to = self.model_options["transformer_options"]