Fix: Safe handling for multimodal_config to avoid 'NoneType' object h… (#227)

Signed-off-by: Baoyuan Qi <qibaoyuan@126.com>
This commit is contained in:
Baoyuan Qi authored and GitHub committed 2025-12-07 11:40:17 +08:00
1 parent 059f45e48d
commit cfe7ad47c1
1 file changed
+4 -1
+4 -1
View File
@@ -52,7 +52,10 @@ class OmniEngineArgs(EngineArgs):
# and adding the new omni-specific fields
config_dict = base_config.__dict__.copy()
# FIXME(Isotr0py): This is a temporary workaround for multimodal_config
config_dict = {**config_dict.pop("multimodal_config", {}).__dict__, **config_dict}
config_dict = {
**(getattr(mm := config_dict.pop("multimodal_config", None), "__dict__", mm or {})),
**config_dict,
}
# Add the new omni-specific fields
config_dict["stage_id"] = self.stage_id