SDNQ update CosmosTransformer3DModel keys

This commit is contained in:
Disty0
2026-02-04 10:19:23 +03:00
parent bf1414385f
commit f439d51ea7
+1 -1
View File
@@ -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": [