Reformat convert_videomae_to_pytorch using black

huggingface · Apr 15, 2023 · 84647ee · 84647ee
1 parent 4bd8bd3
commit 84647ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/videomae/convert_videomae_to_pytorch.py b/src/transformers/models/videomae/convert_videomae_to_pytorch.py
@@ -248,7 +248,7 @@ def convert_videomae_checkpoint(checkpoint_url, pytorch_dump_folder_path, model_
         expected_slice = torch.tensor([0.0771, 0.0011, -0.3625])
     elif model_name == "videomae-huge-finetuned-kinetics":
         expected_shape = torch.Size([1, 400])
-        expected_slice = torch.tensor([0.2433,  0.1632, -0.4894])
+        expected_slice = torch.tensor([0.2433, 0.1632, -0.4894])
     elif model_name == "videomae-base-short-finetuned-kinetics":
         expected_shape = torch.Size([1, 400])
         expected_slice = torch.tensor([0.6588, 0.0990, -0.2493])