From ddec40ac1642ef83947b4aa8b3c451e81d20ee1d Mon Sep 17 00:00:00 2001 From: BUAADreamer <1428195643@qq.com> Date: Sun, 29 Sep 2024 21:39:37 +0800 Subject: [PATCH] fix style Former-commit-id: 23916d57c1d22653739dbf913d3e427fcb978a15 --- src/llamafactory/model/model_utils/visual.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/llamafactory/model/model_utils/visual.py b/src/llamafactory/model/model_utils/visual.py index 55f045f6..85f386de 100644 --- a/src/llamafactory/model/model_utils/visual.py +++ b/src/llamafactory/model/model_utils/visual.py @@ -108,7 +108,9 @@ def configure_visual_model(config: "PretrainedConfig") -> None: Patches VLMs before loading them. """ model_type = getattr(config, "model_type", None) - if model_type in ["llava", "video_llava"] or "llava_next" in model_type: # required for ds zero3 and valuehead models + if ( + model_type in ["llava", "video_llava"] or "llava_next" in model_type + ): # required for ds zero3 and valuehead models setattr(config, "hidden_size", getattr(config.text_config, "hidden_size", None)) if getattr(config, "is_yi_vl_derived_model", None):