Fix Whisper auto_model mapping fallback for concrete model classes (#4115)
This commit is contained in:
parent
d9089de0f7
commit
9248091500
1 changed files with 2 additions and 2 deletions
|
|
@ -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"):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue