From 1f47a24aa1fbaefcac9d0ffbfa5fff30867a8c92 Mon Sep 17 00:00:00 2001 From: Turetskii Mikhail Date: Fri, 26 Jan 2024 14:52:53 +0300 Subject: [PATCH] fix: corrected misleading log message in save_pretrained function (#28699) --- src/transformers/modeling_utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py index 585b1ac0052..6e2cc2fbd6f 100644 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -2495,8 +2495,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix save_function(shard, os.path.join(save_directory, shard_file)) if index is None: - weights_file_name = SAFE_WEIGHTS_NAME if safe_serialization else WEIGHTS_NAME - path_to_weights = os.path.join(save_directory, _add_variant(weights_file_name, variant)) + path_to_weights = os.path.join(save_directory, weights_name) logger.info(f"Model weights saved in {path_to_weights}") else: save_index_file = SAFE_WEIGHTS_INDEX_NAME if safe_serialization else WEIGHTS_INDEX_NAME