From 838803d0536c0750cfc26d38b39f175dda9df52a Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 23 Jul 2025 17:58:28 +0300
Subject: [PATCH] Update nodes_model_loading.py

---
 nodes_model_loading.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/nodes_model_loading.py b/nodes_model_loading.py
index 62956ba..f37ef1e 100644
--- a/nodes_model_loading.py
+++ b/nodes_model_loading.py
@@ -111,6 +111,9 @@ def standardize_lora_key_format(lora_sd):
             k = k.replace('transformer.', 'diffusion_model.')
         if k.startswith('pipe.dit.'): #unianimate-dit/diffsynth
             k = k.replace('pipe.dit.', 'diffusion_model.')
+        if k.startswith('blocks.'):
+            k = k.replace('blocks.', 'diffusion_model.blocks.')
+        k = k.replace('.default.', '.')
 
         # Fun LoRA format
         if k.startswith('lora_unet__'):