Remove embarrassing debug print() in save_pretrained (#19922)

c766a2d7 · Matt · GitHub · 1e6141c3 · c766a2d7
Unverified Commit c766a2d7 authored Oct 27, 2022 by Matt Committed by GitHub Oct 27, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

src/transformers/modeling_tf_utils.py src/transformers/modeling_tf_utils.py +0 -1

No files found.
--- a/src/transformers/modeling_tf_utils.py
+++ b/src/transformers/modeling_tf_utils.py
@@ -2222,7 +2222,6 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin, TFGenerationMixin, Pu
                    for layer in sorted(shard, key=lambda x: x.name):
                        if "model." in layer.name or len(layer.name.split("/")) == 1:
                            layer_name = layer.name
-                            print(layer_name)
                        else:
                            layer_name = "/".join(layer.name.split("/")[1:])
                        param_dset = shard_file.create_dataset(