remove unecessary print statments

huggingface · Oct 18, 2023 · 330eaf2 · 330eaf2
1 parent 1ed2469
commit 330eaf2
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/transformers/models/vit/convert_vit_timm_to_pytorch.py b/src/transformers/models/vit/convert_vit_timm_to_pytorch.py
@@ -183,7 +183,6 @@ def convert_vit_checkpoint(vit_name, pytorch_dump_folder_path):
     # load HuggingFace model
     if base_model:
         model = ViTModel(config, add_pooling_layer=False).eval()
-        # print(model.state_dict().keys())
     else:
         model = ViTForImageClassification(config).eval()
     model.load_state_dict(state_dict)
@@ -199,8 +198,6 @@ def convert_vit_checkpoint(vit_name, pytorch_dump_folder_path):
 
     if base_model:
         timm_pooled_output = timm_model.forward_features(pixel_values)
-        print(timm_pooled_output)
-        print(outputs.last_hidden_state)
         assert timm_pooled_output.shape == outputs.last_hidden_state.shape
         assert torch.allclose(timm_pooled_output, outputs.last_hidden_state, atol=1e-1)
     else: