Unverified Commit 68fa1e85 authored by lin yudong's avatar lin yudong Committed by GitHub
Browse files

update the logger message with accordant weights_file_name (#28181)


Co-authored-by: default avataryudong.lin <yudong.lin@funplus.com>
parent 74d9d0ce
...@@ -2408,7 +2408,8 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix ...@@ -2408,7 +2408,8 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
save_function(shard, os.path.join(save_directory, shard_file)) save_function(shard, os.path.join(save_directory, shard_file))
if index is None: if index is None:
path_to_weights = os.path.join(save_directory, _add_variant(WEIGHTS_NAME, variant)) 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))
logger.info(f"Model weights saved in {path_to_weights}") logger.info(f"Model weights saved in {path_to_weights}")
else: else:
save_index_file = SAFE_WEIGHTS_INDEX_NAME if safe_serialization else WEIGHTS_INDEX_NAME save_index_file = SAFE_WEIGHTS_INDEX_NAME if safe_serialization else WEIGHTS_INDEX_NAME
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment