Merge branch 'main' into refactor/model-manager-3

2026-02-13 22:14:59 -05:00 · 2023-11-27 22:15:51 -05:00
parent a79e814c8d d971c5fa64
commit ecd3dcd5df
8 changed files with 142 additions and 163 deletions
--- a/invokeai/backend/model_management/model_probe.py
+++ b/invokeai/backend/model_management/model_probe.py
@@ -238,7 +238,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)