hxssgaa commited on
Commit
6030c5c
·
verified ·
1 Parent(s): c270103

Fix Transformers 5 tied-weight save mapping

Browse files

Keep the nested Qwen3-VL tied-weight target-to-source mapping as a dictionary so Transformers 5 can save the model and Sentence Transformers MultiVectorEncoder.

Files changed (1) hide show
  1. modeling_colqwen3.py +4 -1
modeling_colqwen3.py CHANGED
@@ -139,7 +139,10 @@ class ColQwen3(ColQwen3PreTrainedModel):
139
  )
140
  self.padding_side = getattr(config, "padding_side", "left")
141
  self.mask_non_image_embeddings = mask_non_image_embeddings
142
- self._tied_weights_keys = [f"vlm.{k}" for k in (self.vlm._tied_weights_keys or [])]
 
 
 
143
 
144
  self.post_init()
145
 
 
139
  )
140
  self.padding_side = getattr(config, "padding_side", "left")
141
  self.mask_non_image_embeddings = mask_non_image_embeddings
142
+ inner_tied = self.vlm._tied_weights_keys or {}
143
+ self._tied_weights_keys = {
144
+ f"vlm.{target}": f"vlm.{source}" for target, source in inner_tied.items()
145
+ }
146
 
147
  self.post_init()
148