"test/srt/vscode:/vscode.git/clone" did not exist on "995af5a54b03495ff34af28f5499f381d19758da"
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:
model_class ([`PreTrainedModel`]):
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(
"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 "
......
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