Print text_input_parts which cause cuda error.

utils/model.py

@@ -417,6 +417,8 @@ class OwlViTForClassification(nn.Module):
             topk_scores, topk_idxs = torch.topk(teacher_boxes_logits, k=1, dim=1)
 
         else:
+            #DEUBUG:
+            print(f"text_inputs_parts: {text_inputs_parts}")
             text_embeds_parts = self.owlvit.get_text_features(**text_inputs_parts)
             
             # # Embed images and text queries