Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Lincoln Stein
2023-11-27 22:15:51 -05:00
committed by GitHub
parent a79e814c8d d971c5fa64
commit ecd3dcd5df
8 changed files with 142 additions and 163 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_management/model_probe.py
View File

@ -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)