From 8ad53ed4b3748132cb1c58d0c60d1eb641740cdf Mon Sep 17 00:00:00 2001
From: Disty0 <disty@disty.xyz>
Date: Tue, 4 Nov 2025 14:29:44 +0300
Subject: [PATCH 1/2] SDNQ update keys

---
 modules/sdnq/common.py | 37 ++++++++++++++++++++++---------------
 1 file changed, 22 insertions(+), 15 deletions(-)

diff --git a/modules/sdnq/common.py b/modules/sdnq/common.py
index e83af542b..410c58216 100644
--- a/modules/sdnq/common.py
+++ b/modules/sdnq/common.py
@@ -83,10 +83,7 @@ else:
 
 
 common_skip_keys = (
-    ".lm_head",
-    ".patch_embedding",
     ".time_embed",
-    ".time_text_embed",
     ".context_embedder",
     ".condition_embedder",
     ".x_embedder",
@@ -101,39 +98,48 @@ common_skip_keys = (
     ".img_out",
     ".vid_out",
     ".final_layer",
-    "pos_embed",
+    "time_text_embed",
+    "patch_embedding",
+    "patch_embed",
+    "patch_emb",
+    "lm_head",
+    "wte",
 )
 
 module_skip_keys_dict = {
     "FluxTransformer2DModel": [
-        ["single_transformer_blocks.0.norm.linear.weight", ".time_text_embed", ".context_embedder", ".x_embedder", ".proj_out", ".norm_out", "pos_embed"],
+        ["single_transformer_blocks.0.norm.linear.weight", "time_text_embed", "context_embedder", "x_embedder", ".proj_out", "norm_out"],
         {}
     ],
     "ChromaTransformer2DModel": [
-        ["distilled_guidance_layer", ".time_text_embed", ".context_embedder", ".x_embedder", ".proj_out", ".norm_out", "pos_embed"],
+        ["distilled_guidance_layer", "time_text_embed", "context_embedder", "x_embedder", ".proj_out", "norm_out"],
         {}
     ],
     "QwenImageTransformer2DModel": [
-        ["transformer_blocks.0.img_mod.1.weight", ".time_text_embed", ".txt_in", ".img_in", ".proj_out", ".norm_out", "pos_embed"],
+        ["transformer_blocks.0.img_mod.1.weight", "time_text_embed", "txt_in", "img_in", "proj_out", "norm_out"],
         {}
     ],
     "WanTransformer3DModel": [
-        ["scale_shift_table", ".rope", ".patch_embedding", ".condition_embedder", ".proj_out", ".norm_out", "pos_embed"],
-        {}
-    ],
-    "Gemma3nForCausalLM": [
-        [".lm_head", "correction_coefs", "prediction_coefs", "embedding_projection"],
+        ["scale_shift_table", "patch_embedding", "condition_embedder", "proj_out", "norm_out"],
         {}
     ],
     "HunyuanImage3ForCausalMM": [
-        [".lm_head", ".patch_embed", ".time_embed", ".time_embed_2", ".final_layer", ".model.wte", ".model.ln_f", ".timestep_emb", ".vae", ".vision_aligner", ".vision_model.head", ".vision_model.post_layernorm", ".vision_model.embeddings"],
+        ["lm_head", "patch_embed", "time_embed", "time_embed_2", "final_layer", "wte", "ln_f", "timestep_emb", "vae", "vision_aligner", "head", "post_layernorm", "embeddings"],
         {}
     ],
     "Emu3ForCausalLM": [
-        [".lm_head", ".vq_model", ".tokenizer"],
+        ["lm_head", "vq_model", "tokenizer"],
         {}
     ],
-    "NaDiT": [
+    "Gemma3nForCausalLM": [
+        ["lm_head", "correction_coefs", "prediction_coefs", "embedding_projection"],
+        {}
+    ],
+    "MoondreamModel": [
+        ["lm_head", "region", "wte", "post_ln", "proj_mlp", "patch_emb", "pos_emb"],
+        {}
+    ],
+    "D": [
         [".emb_in", ".txt_in", ".vid_in", ".emb_scale", ".vid_out", ".vid_out_norm", ".vid_out_ada"],
         {}
     ],
@@ -141,4 +147,5 @@ module_skip_keys_dict = {
 
 module_skip_keys_dict["ChronoEditTransformer3DModel"] = module_skip_keys_dict["WanTransformer3DModel"]
 module_skip_keys_dict["Gemma3nForConditionalGeneration"] = module_skip_keys_dict["Gemma3nForCausalLM"]
+module_skip_keys_dict["HfMoondream"] = module_skip_keys_dict["MoondreamModel"]
 module_skip_keys_dict["NaDiTUpscaler"] = module_skip_keys_dict["NaDiT"]

From a4378a79e4683be50860df1e554ac5a2338bdf4b Mon Sep 17 00:00:00 2001
From: Disty0 <disty@disty.xyz>
Date: Tue, 4 Nov 2025 14:30:52 +0300
Subject: [PATCH 2/2] fix typo

---
 modules/sdnq/common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/sdnq/common.py b/modules/sdnq/common.py
index 410c58216..61d52d110 100644
--- a/modules/sdnq/common.py
+++ b/modules/sdnq/common.py
@@ -139,7 +139,7 @@ module_skip_keys_dict = {
         ["lm_head", "region", "wte", "post_ln", "proj_mlp", "patch_emb", "pos_emb"],
         {}
     ],
-    "D": [
+    "NaDiT": [
         [".emb_in", ".txt_in", ".vid_in", ".emb_scale", ".vid_out", ".vid_out_norm", ".vid_out_ada"],
         {}
     ],