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