From c56594b0b82628b9c8d61242afbd0559544dd635 Mon Sep 17 00:00:00 2001 From: Matt Date: Thu, 9 Jan 2025 14:33:20 +0000 Subject: [PATCH 1/2] Fix the config class comparison when repeatedly saving and loading remote code models --- src/transformers/models/auto/auto_factory.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/transformers/models/auto/auto_factory.py b/src/transformers/models/auto/auto_factory.py index 7809b2a6cc2c..69b999c75342 100644 --- a/src/transformers/models/auto/auto_factory.py +++ b/src/transformers/models/auto/auto_factory.py @@ -580,7 +580,8 @@ def register(cls, config_class, model_class, exist_ok=False): model_class ([`PreTrainedModel`]): The model to register. """ - if hasattr(model_class, "config_class") and str(model_class.config_class) != str(config_class): + if hasattr(model_class, "config_class") and model_class.config_class.__name__ != config_class.__name__: + breakpoint() 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 " From 0bb5b325b555aa0bbd07302b8556727fdd8b304d Mon Sep 17 00:00:00 2001 From: Matt Date: Thu, 9 Jan 2025 14:36:40 +0000 Subject: [PATCH 2/2] once again you have committed your debug breakpoint --- src/transformers/models/auto/auto_factory.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/transformers/models/auto/auto_factory.py b/src/transformers/models/auto/auto_factory.py index 69b999c75342..069269bb7f70 100644 --- a/src/transformers/models/auto/auto_factory.py +++ b/src/transformers/models/auto/auto_factory.py @@ -581,7 +581,6 @@ def register(cls, config_class, model_class, exist_ok=False): The model to register. """ if hasattr(model_class, "config_class") and model_class.config_class.__name__ != config_class.__name__: - breakpoint() 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 "