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

allow submodels to be fetched independent of parent pipeline

Browse Source
This commit is contained in:
Lincoln Stein 2023-05-13 16:54:47 -04:00
parent d8b1f29066
commit 2204e47596
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/backend/model_management/model_manager.py
View File

@ -357,6 +357,13 @@ class ModelManager(object):
vae_id = mconfig.vae.repo_id vae_id = mconfig.vae.repo_id
vae = (SDModelType.vae, vae_id) vae = (SDModelType.vae, vae_id)
# optimization - don't load whole model if the user
# is asking for just a piece of it
if model_type == SDModelType.diffusers and submodel and not subfolder:
model_type = submodel
subfolder = submodel.name
submodel = None
model_context = self.cache.get_model( model_context = self.cache.get_model(
location, location,
model_type = model_type, model_type = model_type,