Skip to content

Commit

Permalink
Fix regression on Processor.save_pretrained caused by huggingface#3…
Browse files Browse the repository at this point in the history
…1691 (huggingface#32921)

fix save_pretrained
  • Loading branch information
leloykun authored and itazap committed Sep 20, 2024
1 parent e73fa01 commit 21b722f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/processing_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]

Expand Down

0 comments on commit 21b722f

Please sign in to comment.