Update utils/model.py

utils/model.py

@@ -418,7 +418,7 @@ class OwlViTForClassification(nn.Module):
 
         else:
             #DEUBUG:
-            print(f"text_inputs_parts: {text_inputs_parts}")
+            print(f"text_inputs_parts - input_ids: {text_inputs_parts['input_ids'].shape}. attention_mask : {text_inputs_parts['attention_mask'].shape}")
             text_embeds_parts = self.owlvit.get_text_features(**text_inputs_parts)
             
             # # Embed images and text queries