Skip to content

Commit cdf04ff

Browse files
zucchini-nlpArthurZucker
authored andcommitted
[qwen-vl] Look for vocab size in text config (#38372)
fix qwen
1 parent 2842b82 commit cdf04ff

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1797,7 +1797,7 @@ class Qwen2_5_VLForConditionalGeneration(Qwen2_5_VLPreTrainedModel, GenerationMi
17971797
def __init__(self, config):
17981798
super().__init__(config)
17991799
self.model = Qwen2_5_VLModel(config)
1800-
self.lm_head = nn.Linear(config.hidden_size, config.vocab_size, bias=False)
1800+
self.lm_head = nn.Linear(config.text_config.hidden_size, config.text_config.vocab_size, bias=False)
18011801

18021802
self.post_init()
18031803

src/transformers/models/qwen2_vl/modeling_qwen2_vl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1673,7 +1673,7 @@ class Qwen2VLForConditionalGeneration(Qwen2VLPreTrainedModel, GenerationMixin):
16731673
def __init__(self, config):
16741674
super().__init__(config)
16751675
self.model = Qwen2VLModel(config)
1676-
self.lm_head = nn.Linear(config.hidden_size, config.vocab_size, bias=False)
1676+
self.lm_head = nn.Linear(config.text_config.hidden_size, config.text_config.vocab_size, bias=False)
16771677

16781678
self.post_init()
16791679

0 commit comments

Comments
 (0)