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

fix(install): do not use deprecated pydantic methods

Browse Source
This commit is contained in:
psychedelicious 2024-03-11 23:31:29 +11:00
parent 3d1f3818cb
commit b4b0af7c60
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/install/install_helper.py
View File

@ -149,7 +149,7 @@ class InstallHelper(object):
""" """
# previously-installed models # previously-installed models
for model in self._installer.record_store.all_models(): for model in self._installer.record_store.all_models():
info = UnifiedModelInfo.parse_obj(model.dict()) info = UnifiedModelInfo.model_validate(model.model_dump())
info.installed = True info.installed = True
model_key = f"{model.base.value}/{model.type.value}/{model.name}" model_key = f"{model.base.value}/{model.type.value}/{model.name}"
self.all_models[model_key] = info self.all_models[model_key] = info
@ -183,7 +183,7 @@ class InstallHelper(object):
# previously-installed models # previously-installed models
for model in self._installer.record_store.all_models(): for model in self._installer.record_store.all_models():
info = UnifiedModelInfo.parse_obj(model.dict()) info = UnifiedModelInfo.model_validate(model.model_dump())
info.installed = True info.installed = True
model_key = f"{model.base.value}/{model.type.value}/{model.name}" model_key = f"{model.base.value}/{model.type.value}/{model.name}"
self.all_models[model_key] = info self.all_models[model_key] = info