Skip to content

Commit cd5211b

Browse files
authored
Merge pull request #30 from yiyexy/main
fix a bug
2 parents 5dd1b38 + 72327ec commit cd5211b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ds/merge_model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ def load_empty_model(llm_path):
4444
llava_ov_config = Llavaonevision1_5Config()
4545
llm_config = AutoConfig.from_pretrained(llm_path, trust_remote_code=True)
4646
llava_ov_config.text_config.update(llm_config.to_dict())
47+
llava_ov_config.vision_config.text_hidden_size = llava_ov_config.text_config.hidden_size
4748
model = LLaVAOneVision1_5_ForConditionalGeneration(llava_ov_config)
4849
return model, processor, tokenizer
4950

0 commit comments

Comments
 (0)