diff --git a/nodes_model_loading.py b/nodes_model_loading.py index dfe0947..d0ee8f9 100644 --- a/nodes_model_loading.py +++ b/nodes_model_loading.py @@ -128,6 +128,8 @@ def standardize_lora_key_format(lora_sd): k = k.replace('pipe.dit.', 'diffusion_model.') if k.startswith('blocks.'): k = k.replace('blocks.', 'diffusion_model.blocks.') + if k.startswith('vace_blocks.'): + k = k.replace('vace_blocks.', 'diffusion_model.vace_blocks.') k = k.replace('.default.', '.') # Fun LoRA format