github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

docs(mm): add reminder for self for field migrations

Browse Source
This commit is contained in:
psychedelicious
2025-10-07 19:35:07 +11:00
parent 01ca74e622
commit 99d3f16eb4
2 changed files with 23 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
invokeai/app/services/shared/sqlite_migrator/migrations/migration_22.py
View File

@@ -2,7 +2,7 @@ import json
import sqlite3
from logging import Logger
from pathlib import Path
from typing import NamedTuple
from typing import Any, NamedTuple
from pydantic import ValidationError
@@ -29,8 +29,9 @@ class Migration22Callback:
for model_id, config_json in rows:
try:
migrated_config_dict = self._migrate_config(config_json)
# Get the model config as a pydantic object
config = AnyModelConfigValidator.validate_json(config_json)
config = AnyModelConfigValidator.validate_python(migrated_config_dict)
except ValidationError:
# This could happen if the config schema changed in a way that makes old configs invalid. Unlikely
# for users, more likely for devs testing out migration paths.
@@ -76,6 +77,26 @@ class Migration22Callback:
self._prune_empty_directories()
def _migrate_config(self, config_json: Any) -> str | None:
config_dict = json.loads(config_json)
# TODO: migrate fields, review changes to ensure we hit all cases for v6.7.0 to v6.8.0 upgrade.
# Prior to v6.8.0, we used an awkward combination of `config_path` and `variant` to distinguish between FLUX
# variants.
#
# `config_path` was set to one of:
# - flux-dev
# - flux-dev-fill
# - flux-schnell
#
# `variant` was set to ModelVariantType.Inpaint for FLUX Fill models and ModelVariantType.Normal for all other FLUX
# models.
#
# We now use the `variant` field to directly represent the FLUX variant type, and `config_path` is no longer used.
return config_dict
def _normalize_model_storage(self, key: str, path_value: str) -> NormalizeResult:
models_dir = self._models_dir
stored_path = Path(path_value)

12
invokeai/backend/model_manager/configs/main.py
View File

@@ -264,18 +264,6 @@ class Main_Checkpoint_FLUX_Config(Checkpoint_Config_Base, Main_Config_Base, Conf
base: Literal[BaseModelType.Flux] = Field(default=BaseModelType.Flux)
variant: FluxVariantType = Field()
# Prior to v6.8.0, we used an awkward combination of `config_path` and `variant` to distinguish between FLUX
# variants.
#
# `config_path` was set to one of:
# - flux-dev
# - flux-dev-fill
# - flux-schnell
#
# `variant` was set to ModelVariantType.Inpaint for FLUX Fill models and ModelVariantType.Normal for all other FLUX
# models.
#
# We now use the `variant` field to directly represent the FLUX variant type, and `config_path` is no longer used.
@classmethod
def from_model_on_disk(cls, mod: ModelOnDisk, override_fields: dict[str, Any]) -> Self: