Update utils/model.py

utils/model.py

@@ -424,7 +424,9 @@ class OwlViTForClassification(nn.Module):
             txt_embeds = self.owlvit.text_model.embeddings.token_embedding(text_inputs_parts['input_ids'])
             print(f"position_embedding: {self.owlvit.text_model.embeddings.position_embedding(position_ids).shape}")
             print(f"text_embeds: {txt_embeds.shape}")
-            
+
+            device_ = txt_embeds.device
+            position_ids = position_ids.to(device_)
             text_inputs_parts["position_ids"] = position_ids
             position_ids = position_ids.repeat(1, 1, txt_embeds.size(-1) // position_ids.size(-1))
             print(f"pos + emb: {(txt_embeds + position_ids).shape}")