Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

chore: always use cpu as default device

Browse Source 
this is overridden anyways
This commit is contained in:
Manuel Schmid
2024-01-26 02:34:25 +01:00
parent 486c5b4fa5
commit 62fe86f1e8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extras/GroundingDINO/util/inference.py
+1 -1
View File
@@ -16,7 +16,7 @@ class GroundingDinoModel(Model):
def __init__(self): def __init__(self):
self.config_file = 'extras/GroundingDINO/config/GroundingDINO_SwinT_OGC.py' self.config_file = 'extras/GroundingDINO/config/GroundingDINO_SwinT_OGC.py'
self.model = None 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') self.offload_device = torch.device('cpu')
def predict_with_caption( def predict_with_caption(