Yaphabates commited on
Commit
1684a2b
·
verified ·
1 Parent(s): 0ff2bf3

Fix: AutoModelForImageTextToText, arXiv preprint, _tied_weights_keys dict

Browse files
Files changed (1) hide show
  1. modeling_nas_child_vl.py +1 -1
modeling_nas_child_vl.py CHANGED
@@ -28,7 +28,7 @@ class Qwen3VLCausalLMOutputWithPast(CausalLMOutputWithPast):
28
  class NasChildVLModelForCausalLM(Qwen3VLPreTrainedModel, GenerationMixin):
29
  config_class = NasChildVLConfig
30
  _checkpoint_conversion_mapping = {}
31
- _tied_weights_keys = ["lm_head.weight"]
32
 
33
  get_image_features = Qwen3VLModel.get_image_features
34
  get_video_features = Qwen3VLModel.get_video_features
 
28
  class NasChildVLModelForCausalLM(Qwen3VLPreTrainedModel, GenerationMixin):
29
  config_class = NasChildVLConfig
30
  _checkpoint_conversion_mapping = {}
31
+ _tied_weights_keys = {"lm_head.weight": "embed_tokens.weight"}
32
 
33
  get_image_features = Qwen3VLModel.get_image_features
34
  get_video_features = Qwen3VLModel.get_video_features