Unverified Commit d71f5b3e authored by Matt's avatar Matt Committed by GitHub
Browse files

Update config class check in auto factory (#29854)

parent f569172f
...@@ -579,7 +579,7 @@ class _BaseAutoModelClass: ...@@ -579,7 +579,7 @@ class _BaseAutoModelClass:
model_class ([`PreTrainedModel`]): model_class ([`PreTrainedModel`]):
The model to register. The model to register.
""" """
if hasattr(model_class, "config_class") and model_class.config_class != config_class: if hasattr(model_class, "config_class") and str(model_class.config_class) != str(config_class):
raise ValueError( raise ValueError(
"The model class you are passing has a `config_class` attribute that is not consistent with the " "The model class you are passing has a `config_class` attribute that is not consistent with the "
f"config class you passed (model has {model_class.config_class} and you passed {config_class}. Fix " f"config class you passed (model has {model_class.config_class} and you passed {config_class}. Fix "
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment