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

make the convert VAE available to model manager for use in UI

Browse Source
This commit is contained in:
Lincoln Stein 2023-07-19 09:05:24 -04:00
parent 5ae2fb0d2b
commit 0b6ef7eb7d
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -938,20 +938,29 @@ class ModelManager(object):
def models_found(self): def models_found(self):
return self.new_models_found return self.new_models_found
config = self.app_config
# LS: hacky
# Patch in the SD VAE from core so that it is available for use by the UI
try:
self.heuristic_import({config.root_path / 'models/core/convert/sd-vae-ft-mse'})
except:
pass
installer = ModelInstall(config = self.app_config, installer = ModelInstall(config = self.app_config,
model_manager = self, model_manager = self,
prediction_type_helper = ask_user_for_prediction_type, prediction_type_helper = ask_user_for_prediction_type,
) )
config = self.app_config
known_paths = {config.root_path / x['path'] for x in self.list_models()} known_paths = {config.root_path / x['path'] for x in self.list_models()}
directories = {config.root_path / x for x in [config.autoimport_dir, directories = {config.root_path / x for x in [config.autoimport_dir,
config.lora_dir, config.lora_dir,
config.embedding_dir, config.embedding_dir,
config.controlnet_dir] config.controlnet_dir,
]
} }
scanner = ScanAndImport(directories, self.logger, ignore=known_paths, installer=installer) scanner = ScanAndImport(directories, self.logger, ignore=known_paths, installer=installer)
scanner.search() scanner.search()
return scanner.models_found() return scanner.models_found()
def heuristic_import(self, def heuristic_import(self,