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): rename ckpt "last_modified" -> "converted_at"

Browse Source 
Clarify what this timestamp means
This commit is contained in:
psychedelicious
2024-03-01 15:27:41 +11:00
parent 9b40c28144
commit bd4fd9693d
5 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -138,9 +138,7 @@ class ModelConfigBase(BaseModel):
@staticmethod
def json_schema_extra(schema: dict[str, Any], model_class: Type[BaseModel]) -> None:
schema["required"].extend(
["key", "base", "type", "format", "original_hash", "current_hash", "source", "last_modified"]
)
schema["required"].extend(["key", "base", "type", "format", "original_hash", "current_hash", "source"])
model_config = ConfigDict(
use_enum_values=False,
@ -159,7 +157,7 @@ class CheckpointConfigBase(ModelConfigBase):
format: Literal[ModelFormat.Checkpoint] = ModelFormat.Checkpoint
config_path: str = Field(description="path to the checkpoint model config file")
last_modified: Optional[float] = Field(
converted_at: Optional[float] = Field(
description="When this model was last converted to diffusers", default_factory=time.time
)
@ -380,5 +378,5 @@ class ModelConfigFactory(object):
if key:
model.key = key
if isinstance(model, CheckpointConfigBase) and timestamp is not None:
model.last_modified = timestamp
model.converted_at = timestamp
return model # type: ignore