Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "9cf4a8b45668cbcb1bb10ad61d9a9c2c27ab9890"
Unverified Commit 658e5d8f authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

make daily CI happy (#20410)

parent 81c46679
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/transformers/models/owlvit/modeling_owlvit.py
View file @ 658e5d8f
...@@ -1475,9 +1475,10 @@ class OwlViTForObjectDetection(OwlViTPreTrainedModel): ...@@ -1475,9 +1475,10 @@ class OwlViTForObjectDetection(OwlViTPreTrainedModel):
# Loop over query images # Loop over query images
best_class_embeds = [] best_class_embeds = []
best_box_indices = [] best_box_indices = []
pred_boxes_device = pred_boxes_as_corners.device
for i in range(query_image_features.shape[0]): for i in range(query_image_features.shape[0]):
each_query_box = torch.tensor([[0, 0, 1, 1]]) each_query_box = torch.tensor([[0, 0, 1, 1]], device=pred_boxes_device)
each_query_pred_boxes = pred_boxes_as_corners[i] each_query_pred_boxes = pred_boxes_as_corners[i]
ious, _ = box_iou(each_query_box, each_query_pred_boxes) ious, _ = box_iou(each_query_box, each_query_pred_boxes)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment