From f439d51ea772e89ad404856fd2cc6fd6344af143 Mon Sep 17 00:00:00 2001
From: Disty0 <disty@disty.xyz>
Date: Wed, 4 Feb 2026 10:19:23 +0300
Subject: [PATCH] SDNQ update CosmosTransformer3DModel keys

---
 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 58e8b9c03..70dc2a8ce 100644
--- a/modules/sdnq/common.py
+++ b/modules/sdnq/common.py
@@ -315,7 +315,7 @@ module_skip_keys_dict = {
         {}
     ],
     "CosmosTransformer3DModel": [
-        ["patch_embed", "time_embed", "norm_out", "proj_out", "crossattn_proj"],
+        ["transformer_blocks.0.norm*", "patch_embed", "time_embed", "norm_out", "proj_out", "crossattn_proj"],
         {}
     ],
     "GlmImageTransformer2DModel": [