fixes

Author: Aravind-11

src/transformers/models/owlvit/modeling_owlvit.py

@@ -401,7 +401,7 @@ def eager_attention_forward(
 class OwlViTAttention(nn.Module):
     """Multi-headed attention from 'Attention Is All You Need' paper"""
 
-    def __init__(self, config: Union[OwlViTVisionConfig, OwlViTTextConfig]):
+    def __init__(self, config):
         super().__init__()
         self.config = config
         self.embed_dim = config.hidden_size