force config intergrity

2024-08-03 17:09:09 -07:00
parent be3f0a0039
commit 6dd8cd8820
1 changed files with 1 additions and 0 deletions
@@ -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