Skip to content

Commit 7f144a0

Browse files
committed
make fixup
1 parent dde8fe5 commit 7f144a0

File tree

3 files changed

+7
-4
lines changed

3 files changed

+7
-4
lines changed

src/transformers/processing_utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -625,7 +625,9 @@ def save_pretrained(self, save_directory, push_to_hub: bool = False, **kwargs):
625625
# plus we save chat_template in its own file
626626
output_processor_file = os.path.join(save_directory, PROCESSOR_NAME)
627627
output_chat_template_file_jinja = os.path.join(save_directory, CHAT_TEMPLATE_FILE)
628-
output_chat_template_file_legacy = os.path.join(save_directory, LEGACY_PROCESSOR_CHAT_TEMPLATE_FILE) # Legacy filename
628+
output_chat_template_file_legacy = os.path.join(
629+
save_directory, LEGACY_PROCESSOR_CHAT_TEMPLATE_FILE
630+
) # Legacy filename
629631
chat_template_dir = os.path.join(save_directory, CHAT_TEMPLATE_DIR)
630632

631633
processor_dict = self.to_dict()

src/transformers/utils/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -72,10 +72,13 @@
7272
working_or_temp_dir,
7373
)
7474
from .hub import (
75+
CHAT_TEMPLATE_DIR,
76+
CHAT_TEMPLATE_FILE,
7577
CLOUDFRONT_DISTRIB_PREFIX,
7678
HF_MODULES_CACHE,
7779
HUGGINGFACE_CO_PREFIX,
7880
HUGGINGFACE_CO_RESOLVE_ENDPOINT,
81+
LEGACY_PROCESSOR_CHAT_TEMPLATE_FILE,
7982
PYTORCH_PRETRAINED_BERT_CACHE,
8083
PYTORCH_TRANSFORMERS_CACHE,
8184
S3_BUCKET_PREFIX,
@@ -98,9 +101,6 @@
98101
list_repo_templates,
99102
send_example_telemetry,
100103
try_to_load_from_cache,
101-
LEGACY_PROCESSOR_CHAT_TEMPLATE_FILE,
102-
CHAT_TEMPLATE_FILE,
103-
CHAT_TEMPLATE_DIR,
104104
)
105105
from .import_utils import (
106106
ACCELERATE_MIN_VERSION,

src/transformers/utils/hub.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@
7171
is_training_run_on_sagemaker,
7272
)
7373

74+
7475
LEGACY_PROCESSOR_CHAT_TEMPLATE_FILE = "chat_template.json"
7576
CHAT_TEMPLATE_FILE = "chat_template.jinja"
7677
CHAT_TEMPLATE_DIR = "additional_chat_templates"

0 commit comments

Comments
 (0)