force config intergrity

This commit is contained in:
layerdiffusion
2024-08-03 17:09:09 -07:00
parent be3f0a0039
commit 6dd8cd8820

View File

@@ -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