fix csm

ArthurZucker · ArthurZucker · commit 6d89354ec21d · 2025-11-13T16:59:19.000+01:00
diff --git a/utils/check_config_attributes.py b/utils/check_config_attributes.py
@@ -310,6 +310,7 @@
     "Gemma3nVisionConfig": ["architecture", "do_pooling", "model_args"],  # this is for use in `timm`
     "VaultGemmaConfig": ["tie_word_embeddings"],
     "GemmaConfig": ["tie_word_embeddings"],
+    "CsmConfig": ["tie_codebooks_embeddings"],
 }
 
 

Original file line number	Diff line number	Diff line change
`@@ -310,6 +310,7 @@`
`310`	`310`	"Gemma3nVisionConfig": ["architecture", "do_pooling", "model_args"], # this is for use in `timm`
`311`	`311`	`"VaultGemmaConfig": ["tie_word_embeddings"],`
`312`	`312`	`"GemmaConfig": ["tie_word_embeddings"],`
	`313`	`+ "CsmConfig": ["tie_codebooks_embeddings"],`
`313`	`314`	`}`
`314`	`315`
`315`	`316`