423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-23 06:12:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update visual.py

Browse Source 
Former-commit-id: 0baa7735f64cbef9bd90e1db485c120b4c1c88bd
This commit is contained in:
hoshi-hiyouga 2024-10-29 22:10:29 +08:00 committed by GitHub
parent fb8f35558a
commit 2179b91acb
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/llamafactory/model/model_utils/visual.py
View File

@ -113,6 +113,7 @@ def configure_visual_model(config: "PretrainedConfig") -> None:
"llava_next", "llava_next",
"llava_next_video", "llava_next_video",
"paligemma", "paligemma",
"pixtral",
"video_llava", "video_llava",
]: # required for ds zero3 and valuehead models ]: # required for ds zero3 and valuehead models
setattr(config, "hidden_size", getattr(config.text_config, "hidden_size", None)) setattr(config, "hidden_size", getattr(config.text_config, "hidden_size", None))