SDNQ clamp weight_std and add CosmosTransformer3DModel keys

2026-09-19 17:24:32 +02:00 · 2026-02-04 00:31:00 +03:00
parent 40df9ce689
commit df30ba28ac
2 changed files with 5 additions and 1 deletions
@@ -314,6 +314,10 @@ module_skip_keys_dict = {
        ["layers.0.adaLN_modulation.0.weight", "t_embedder", "cap_embedder", "siglip_embedder", "all_x_embedder", "all_final_layer"],
        {}
    ],
+    "CosmosTransformer3DModel": [
+        ["patch_embed", "time_embed", "norm_out", "proj_out", "crossattn_proj"],
+        {}
+    ],
    "GlmImageTransformer2DModel": [
        ["transformer_blocks.0.norm1.linear.weight", "image_projector", "glyph_projector", "prior_projector", "time_condition_embed", "norm_out", "proj_out"],
        {}