chore: always use cpu as default device

this is overridden anyways
2024-01-26 02:34:25 +01:00 · 2024-01-26 02:34:25 +01:00 · 62fe86f1e8
parent 486c5b4fa5
commit 62fe86f1e8
1 changed files with 1 additions and 1 deletions
--- a/extras/GroundingDINO/util/inference.py
+++ b/extras/GroundingDINO/util/inference.py
@ -16,7 +16,7 @@ class GroundingDinoModel(Model):
    def __init__(self):
        self.config_file = 'extras/GroundingDINO/config/GroundingDINO_SwinT_OGC.py'
        self.model = None
-        self.load_device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+        self.load_device = torch.device('cpu')
        self.offload_device = torch.device('cpu')

    def predict_with_caption(