From c33e75d3a323218ae732e61e999e92bcf6a3cb08 Mon Sep 17 00:00:00 2001
From: Zhijie Lin <136380243@qq.com>
Date: Thu, 25 May 2023 17:23:32 +0800
Subject: [PATCH] test

---
 GroundingModel.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/GroundingModel.py b/GroundingModel.py
index f14b844..555e12b 100644
--- a/GroundingModel.py
+++ b/GroundingModel.py
@@ -54,7 +54,7 @@ class GroundingModule(nn.Module):
             prompt = prompt + "."
         _, image_tensor = image_transform_grounding(original_image)
         boxes, logits, phrases = predict(self.grounding_model,
-                                         image_tensor, prompt, box_threshold, text_threshold, device=self.device)
+                                         image_tensor, prompt, box_threshold, text_threshold, device='cpu')
         print(phrases)
         # from PIL import Image, ImageDraw, ImageFont
         H, W = original_image.size[1], original_image.size[0]