[Bugfix] Fix incorrect resolving order for transformers fallback (#15279)

Signed-off-by: Isotr0py <2037008807@qq.com>
This commit is contained in:
Isotr0py 2025-03-21 18:54:08 +08:00 committed by GitHub
parent 47c7126213
commit 84e00adc8a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -418,11 +418,13 @@ class _ModelRegistry:
if not architectures:
logger.warning("No model architectures are specified")
normalized_arch = []
for model in architectures:
if model not in self.models:
model = "TransformersModel"
normalized_arch.append(model)
# filter out support architectures
normalized_arch = list(
filter(lambda model: model in self.models, architectures))
# make sure Transformers fallback are put at the last
if len(normalized_arch) != len(architectures):
normalized_arch.append("TransformersModel")
return normalized_arch
def inspect_model_cls(