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 models.yaml version assertion error in pytests

Browse Source
This commit is contained in:
Lincoln Stein
2023-09-17 17:22:50 -04:00
parent d051c0868e
commit 151ba02022
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_manager/storage/sql.py
View File

@ -78,7 +78,7 @@ class ModelConfigStoreSQL(ModelConfigStore):
finally:
self._lock.release()
assert (
self.version == CONFIG_FILE_VERSION
str(self.version) == CONFIG_FILE_VERSION
), f"Model config version {self.version} does not match expected version {CONFIG_FILE_VERSION}"
def _create_tables(self) -> None:

2
invokeai/backend/model_manager/storage/yaml.py
View File

@ -69,7 +69,7 @@ class ModelConfigStoreYAML(ModelConfigStore):
self._initialize_yaml()
self._config = OmegaConf.load(self._filename)
assert (
self.version == CONFIG_FILE_VERSION
str(self.version) == CONFIG_FILE_VERSION
), f"Model config version {self.version} does not match expected version {CONFIG_FILE_VERSION}"
def _initialize_yaml(self):

2
tests/test_model_manager/configs/relative_sub.models.yaml
View File

@ -1,5 +1,5 @@
__metadata__:
version: 3.1.1
version: 3.2
ed799245c762f6d0a9ddfd4e31fdb010:
name: sdxl-base-1-0
path: sdxl/main/SDXL base 1_0