From db2915f299d777bf4032d50d2a26e741a59d464f Mon Sep 17 00:00:00 2001 From: Franz Louis Cesista Date: Wed, 21 Aug 2024 22:34:37 +0800 Subject: [PATCH] fix save_pretrained --- src/transformers/processing_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/processing_utils.py b/src/transformers/processing_utils.py index 09f62481956e..60085fd00705 100644 --- a/src/transformers/processing_utils.py +++ b/src/transformers/processing_utils.py @@ -522,7 +522,7 @@ def save_pretrained(self, save_directory, push_to_hub: bool = False, **kwargs): token=kwargs.get("token"), ) - if set(self.to_dict().keys()) == {"processor_class"}: + if set(processor_dict.keys()) == {"processor_class"}: return [] return [output_processor_file]