Skip to content

Commit 3881660

Browse files
committed
make fixup
1 parent e812178 commit 3881660

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

src/transformers/utils/hub.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -829,7 +829,9 @@ def push_to_hub(
829829
"""
830830
use_auth_token = deprecated_kwargs.pop("use_auth_token", None)
831831
ignore_metadata_errors = deprecated_kwargs.pop("ignore_metadata_errors", False)
832-
save_raw_chat_template = deprecated_kwargs.pop("save_raw_chat_template", None) # Temporary and only for testing
832+
save_raw_chat_template = deprecated_kwargs.pop(
833+
"save_raw_chat_template", None
834+
) # Temporary and only for testing
833835
if use_auth_token is not None:
834836
warnings.warn(
835837
"The `use_auth_token` argument is deprecated and will be removed in v5 of Transformers. Please use `token` instead.",
@@ -887,7 +889,12 @@ def push_to_hub(
887889

888890
# Save all files.
889891
if save_raw_chat_template:
890-
self.save_pretrained(work_dir, max_shard_size=max_shard_size, safe_serialization=safe_serialization, save_raw_chat_template=True)
892+
self.save_pretrained(
893+
work_dir,
894+
max_shard_size=max_shard_size,
895+
safe_serialization=safe_serialization,
896+
save_raw_chat_template=True,
897+
)
891898
else:
892899
self.save_pretrained(work_dir, max_shard_size=max_shard_size, safe_serialization=safe_serialization)
893900

0 commit comments

Comments
 (0)