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

tidy(mm): remove update method from ModelConfigBase

Browse Source 
It's only used in the soon-to-be-removed model merge logic
This commit is contained in:
psychedelicious
2024-03-01 16:34:21 +11:00
parent a8cd3dfc99
commit 61b737bb9f
2 changed files with 3 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/backend/model_manager/config.py
View File

@ -129,7 +129,6 @@ class ModelConfigBase(BaseModel):
key: str = Field(description="unique key for model", default="<NOKEY>")
hash: Optional[str] = Field(description="original fasthash of model contents", default=None)
description: Optional[str] = Field(description="human readable description of the model", default=None)
source: Optional[str] = Field(description="model original source (path, URL or repo_id)", default=None)
@staticmethod
def json_schema_extra(schema: dict[str, Any], model_class: Type[BaseModel]) -> None:
@ -141,11 +140,6 @@ class ModelConfigBase(BaseModel):
json_schema_extra=json_schema_extra,
)
def update(self, attributes: Dict[str, Any]) -> None:
"""Update the object with fields in dict."""
for key, value in attributes.items():
setattr(self, key, value) # may raise a validation error
class CheckpointConfigBase(ModelConfigBase):
"""Model config for checkpoint-style models."""