Skip to content

GitLab

Unverified Commit c766a2d7 authored by Matt's avatar Matt Committed by GitHub
Browse files

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

parent 1e6141c3
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
src/transformers/modeling_tf_utils.py
View file @ c766a2d7
...@@ -2222,7 +2222,6 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin, TFGenerationMixin, Pu ...@@ -2222,7 +2222,6 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin, TFGenerationMixin, Pu
for layer in sorted(shard, key=lambda x: x.name): for layer in sorted(shard, key=lambda x: x.name):
if "model." in layer.name or len(layer.name.split("/")) == 1: if "model." in layer.name or len(layer.name.split("/")) == 1:
layer_name = layer.name layer_name = layer.name
print(layer_name)
else: else:
layer_name = "/".join(layer.name.split("/")[1:]) layer_name = "/".join(layer.name.split("/")[1:])
param_dset = shard_file.create_dataset( param_dset = shard_file.create_dataset(
......
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