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

Do not assign the result of SpandrelImageToImageModel.load_from_file(...) during probe to ensure that the model is immediately gc'd.

Browse Source
This commit is contained in:
Ryan Dick 2024-07-05 14:05:12 -04:00
parent d09999736c
commit 0ce6ec634d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_manager/probe.py
View File

@ -253,7 +253,7 @@ class ModelProbe(object):
# 2. Spandrel has internal logic to determine a model's type from its state_dict before loading the model. # 2. Spandrel has internal logic to determine a model's type from its state_dict before loading the model.
# This logic is not exposed in spandrel's public API. We could copy the logic here, but then we have to # This logic is not exposed in spandrel's public API. We could copy the logic here, but then we have to
# maintain it, and the risk of false positive detections is higher. # maintain it, and the risk of false positive detections is higher.
_ = SpandrelImageToImageModel.load_from_file(model_path) SpandrelImageToImageModel.load_from_file(model_path)
return ModelType.SpandrelImageToImage return ModelType.SpandrelImageToImage
except spandrel.UnsupportedModelError: except spandrel.UnsupportedModelError:
pass pass