fix bug

2023-05-02 19:41:34 +08:00 · 2023-05-02 19:41:34 +08:00 · a4dcf5d411
parent 0dc5ece5a2
commit a4dcf5d411
1 changed files with 2 additions and 2 deletions
--- a/groundingdino/util/inference.py
+++ b/groundingdino/util/inference.py
@ -154,7 +154,7 @@ class Model:
            caption=caption,
            box_threshold=box_threshold,
            text_threshold=text_threshold, 
-            device=self.model.device)
+            device=self.device)
        source_h, source_w, _ = image.shape
        detections = Model.post_process_result(
            source_h=source_h,
@ -197,7 +197,7 @@ class Model:
            caption=caption,
            box_threshold=box_threshold,
            text_threshold=text_threshold,
-            device=self.model.device)
+            device=self.device)
        source_h, source_w, _ = image.shape
        detections = Model.post_process_result(
            source_h=source_h,