Update probe to always use cpu for loading models

invoke-ai · Apr 3, 2024 · c301735 · c301735
1 parent 7da04b8
commit c301735
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/invokeai/backend/model_manager/probe.py b/invokeai/backend/model_manager/probe.py
@@ -324,7 +324,7 @@ def _scan_and_load_checkpoint(cls, model_path: Path) -> CkptType:
         with SilenceWarnings():
             if model_path.suffix.endswith((".ckpt", ".pt", ".pth", ".bin")):
                 cls._scan_model(model_path.name, model_path)
-                model = torch.load(model_path)
+                model = torch.load(model_path, map_location="cpu")
                 assert isinstance(model, dict)
                 return model
             else: