diff --git a/invokeai/backend/model_management/model_probe.py b/invokeai/backend/model_management/model_probe.py index c83be2a6c8..aebe30f116 100644 --- a/invokeai/backend/model_management/model_probe.py +++ b/invokeai/backend/model_management/model_probe.py @@ -225,7 +225,7 @@ class ModelProbe(object): with SilenceWarnings(): if model_path.suffix.endswith((".ckpt", ".pt", ".bin")): cls._scan_model(model_path, model_path) - return torch.load(model_path) + return torch.load(model_path, map_location="cpu") else: return safetensors.torch.load_file(model_path)