From 4aee524ddfebfd280ba22484a3d283e40c8c31de Mon Sep 17 00:00:00 2001
From: Disty0 <disty@disty.xyz>
Date: Tue, 14 Oct 2025 17:18:58 +0300
Subject: [PATCH] SDNQ add NaDiT keys

---
 modules/sdnq/common.py        | 6 ++++++
 modules/seedvr/config_7b.yaml | 2 +-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/modules/sdnq/common.py b/modules/sdnq/common.py
index 698a35633..c5ff8ab9a 100644
--- a/modules/sdnq/common.py
+++ b/modules/sdnq/common.py
@@ -95,4 +95,10 @@ module_skip_keys_dict = {
         ["transformer_blocks.0.img_mod.1.weight", ".time_text_embed", ".txt_in", ".img_in", ".proj_out", ".norm_out", "pos_embed"],
         {}
     ],
+    "NaDiT": [
+        [".emb_in", ".txt_in", ".vid_in", ".emb_scale", ".vid_out", ".vid_out_norm", ".vid_out_ada"],
+        {}
+    ],
 }
+
+module_skip_keys_dict["NaDiTUpscaler"] = module_skip_keys_dict["NaDiT"]
diff --git a/modules/seedvr/config_7b.yaml b/modules/seedvr/config_7b.yaml
index 61803415d..0e5cb146c 100644
--- a/modules/seedvr/config_7b.yaml
+++ b/modules/seedvr/config_7b.yaml
@@ -8,7 +8,7 @@ dit:
       path:
         - "SeedVR2_VideoUpscaler.src.models.dit.nadit"
         - "SeedVR2_VideoUpscaler.src.models.dit.nadit"
-        - "src.models.dit.nadit"
+        - "modules.seedvr.src.models.dit.nadit"
       name: "NaDiT"
       args: "as_params"
     vid_in_channels: 33