github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py

Browse Source 
Co-authored-by: Ryan Dick <ryanjdick3@gmail.com>
This commit is contained in:
Lincoln Stein
2024-06-15 19:08:29 -04:00
committed by GitHub
parent 6b788bff51
commit 1411fbbd1a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py
View File

@@ -128,11 +128,11 @@ class StableDiffusionDiffusersModel(GenericDiffusersLoader):
load_safety_checker=False,
)
# Proactively load the various submodels into the RAM cache so that we don't have to re-load
# the entire pipeline every time a new submodel is needed.
if not submodel_type:
return pipeline
# Proactively load the various submodels into the RAM cache so that we don't have to re-load
# the entire pipeline every time a new submodel is needed.
for subtype in SubModelType:
if subtype == submodel_type:
continue