From 897f9a05e4aecf54e950c830e0c016dd9ed97827 Mon Sep 17 00:00:00 2001 From: a8nova Date: Sun, 5 May 2024 21:07:27 +0300 Subject: [PATCH] Revert "Add logging to modeling tf utils, will be reverted just for debugging" This reverts commit 774b6b7b1c17b3ce5d7634ade768f2f686cee617. --- src/transformers/modeling_tf_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/modeling_tf_utils.py b/src/transformers/modeling_tf_utils.py index 71dc3c25c74048..d5e17d256869a1 100644 --- a/src/transformers/modeling_tf_utils.py +++ b/src/transformers/modeling_tf_utils.py @@ -2475,7 +2475,7 @@ def save_pretrained( output_model_file = os.path.join(save_directory, weights_name) shards, index = tf_shard_checkpoint(self.weights, max_shard_size, weights_name=weights_name) - logger.info(f"shards {shards}\nindex {index}") + # Clean the folder from a previous save for filename in os.listdir(save_directory): full_filename = os.path.join(save_directory, filename)