update

grounding_model.py

@@ -64,7 +64,7 @@ class GroundingModule(nn.Module):
         print('==> Box grounding with "{}"...'.format(prompt))
         with torch.cuda.amp.autocast(enabled=True):
             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]