update post_process_image_guided_detection (#20521)

src/transformers/models/owlvit/image_processing_owlvit.py

@@ -416,7 +416,7 @@ class OwlViTImageProcessor(BaseImageProcessor):
 
         # Convert from relative [0, 1] to absolute [0, height] coordinates
         img_h, img_w = target_sizes.unbind(1)
-        scale_fct = torch.stack([img_w, img_h, img_w, img_h], dim=1)
+        scale_fct = torch.stack([img_w, img_h, img_w, img_h], dim=1).to(target_boxes.device)
         target_boxes = target_boxes * scale_fct[:, None, :]
 
         # Compute box display alphas based on prediction scores