style fixes

huggingface · Sep 8, 2022 · bb61e30 · bb61e30
1 parent efdca2e
commit bb61e30
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/src/transformers/models/owlvit/modeling_owlvit.py b/src/transformers/models/owlvit/modeling_owlvit.py
@@ -1078,7 +1078,7 @@ def get_image_features(
             last_hidden_state = vision_outputs[0]
             image_features = self.vision_model.post_layernorm(last_hidden_state)
         # Unmodified CLIP embeddings
-        else: 
+        else:
             pooled_output = vision_outputs[1]
             image_features = self.visual_projection(pooled_output)
 

diff --git a/tests/models/owlvit/test_modeling_owlvit.py b/tests/models/owlvit/test_modeling_owlvit.py
@@ -762,7 +762,6 @@ def recursive_check(tuple_object, dict_object):
             dict_inputs = self._prepare_for_class(inputs_dict, model_class)
             check_equivalence(model, tuple_inputs, dict_inputs)
 
-
     @slow
     def test_model_from_pretrained(self):
         for model_name in OWLVIT_PRETRAINED_MODEL_ARCHIVE_LIST[:1]: