[BugFix]: add bagel to MM_PREFIX_LM_MODELS (#36316)
Signed-off-by: princepride <wangzhipeng628@gmail.com>
This commit is contained in:
@@ -1140,6 +1140,7 @@ class ModelConfig:
|
|||||||
return bool(self.hf_config.is_mm_prefix_lm)
|
return bool(self.hf_config.is_mm_prefix_lm)
|
||||||
# fallback to list of known models
|
# fallback to list of known models
|
||||||
MM_PREFIX_LM_MODELS = (
|
MM_PREFIX_LM_MODELS = (
|
||||||
|
"bagel",
|
||||||
"gemma3",
|
"gemma3",
|
||||||
"molmo2",
|
"molmo2",
|
||||||
"paligemma",
|
"paligemma",
|
||||||
|
|||||||
Reference in New Issue
Block a user