Fix Whisper auto_model mapping fallback for concrete model classes (#4115)

This commit is contained in:
Daniel Han 2026-02-26 23:43:57 -08:00 committed by GitHub
commit 9248091500

View file

@ -627,8 +627,8 @@ class FastBaseModel:
# Re-resolve model_class after potential config change
try:
model_class = auto_model._model_mapping[auto_config.__class__]
except KeyError:
pass
except Exception:
model_class = None
model_type = str(getattr(auto_config, "model_type", "")).lower()
if model_type.startswith("gemma3n"):