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

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

diff --git a/GroundingModel.py b/GroundingModel.py
index c767131..f14b844 100644
--- a/GroundingModel.py
+++ b/GroundingModel.py
@@ -35,7 +35,7 @@ class GroundingModule(nn.Module):
         groundingdino_config_file = "./groundingdino/config/GroundingDINO_SwinT_OGC.py"
 
         self.grounding_model = load_groundingdino_model(groundingdino_config_file, groundingdino_checkpoint).to(device)
-        self.sam_predictor = SamPredictor(build_sam(checkpoint=sam_checkpoint)).to(device)
+        self.sam_predictor = SamPredictor(build_sam(checkpoint=sam_checkpoint).to(device))
 
     def prompt2mask(self, original_image, prompt, box_threshold=0.25, text_threshold=0.25, num_boxes=10):
         def image_transform_grounding(init_image):