fix(mimo_v2): auto-disable multimodal when vision/audio configs are absent (#24652)
This commit is contained in:
@@ -226,6 +226,16 @@ class ModelConfig:
|
|||||||
logger.info(
|
logger.info(
|
||||||
f"Multimodal is disabled for {self.hf_config.model_type}. To enable it, set --enable-multimodal."
|
f"Multimodal is disabled for {self.hf_config.model_type}. To enable it, set --enable-multimodal."
|
||||||
)
|
)
|
||||||
|
elif self.hf_config.architectures[0] in MIMO_V2_MULTIMODAL_ARCHS and not (
|
||||||
|
hasattr(self.hf_config, "vision_config")
|
||||||
|
and hasattr(self.hf_config, "audio_config")
|
||||||
|
):
|
||||||
|
enable_multimodal = False
|
||||||
|
logger.info(
|
||||||
|
"Multimodal is disabled for this MiMoV2 checkpoint: "
|
||||||
|
"vision_config/audio_config not found in the model config "
|
||||||
|
"(likely a text-only MiMoV2 variant)."
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
enable_multimodal = True
|
enable_multimodal = True
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user