From 62fe86f1e8f85d47dd5fdfdb79d3379c60061f7d Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Fri, 26 Jan 2024 02:34:25 +0100 Subject: [PATCH] chore: always use cpu as default device this is overridden anyways --- extras/GroundingDINO/util/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extras/GroundingDINO/util/inference.py b/extras/GroundingDINO/util/inference.py index 3f61e946..259094f2 100644 --- 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(