From 6dd8cd882070bbd9e12d74c3b8e0789bd5c8b471 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Sat, 3 Aug 2024 17:09:09 -0700
Subject: [PATCH] force config intergrity

---
 backend/loader.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/backend/loader.py b/backend/loader.py
index 51f6b6bf..f7b4b9d5 100644
--- a/backend/loader.py
+++ b/backend/loader.py
@@ -63,6 +63,7 @@ def load_component(guess, component_name, lib_name, cls_name, repo_path, state_d
 
             with using_forge_operations():
                 model = IntegratedUNet2DConditionModel.from_config(guess.unet_config)
+                model._internal_dict = guess.unet_config
 
             load_state_dict(model, sd)
             return model