diff --git a/invokeai/backend/model_manager/probe.py b/invokeai/backend/model_manager/probe.py index 7aa650afaa..bf21a7fe7b 100644 --- a/invokeai/backend/model_manager/probe.py +++ b/invokeai/backend/model_manager/probe.py @@ -324,7 +324,7 @@ class ModelProbe(object): 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: