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

blackify

Browse Source
This commit is contained in:
Lincoln Stein 2023-09-24 19:00:38 -04:00
parent 593fb95213
commit 912087e4dc
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/backend/install/model_install_backend.py
View File

@ -88,6 +88,7 @@ class ModelLoadInfo:
default: bool = False default: bool = False
requires: Optional[List[str]] = field(default_factory=list) requires: Optional[List[str]] = field(default_factory=list)
class ModelInstall(object): class ModelInstall(object):
def __init__( def __init__(
self, self,
@ -270,7 +271,7 @@ class ModelInstall(object):
additional_models = [] additional_models = []
all_models = self.all_models() all_models = self.all_models()
for path in model_list: for path in model_list:
if not(key := self.reverse_paths.get(path)): if not (key := self.reverse_paths.get(path)):
continue continue
for requirement in all_models[key].requires: for requirement in all_models[key].requires:
requirement_key = self.reverse_paths.get(requirement) requirement_key = self.reverse_paths.get(requirement)