Skip to content

Commit 16570df

Browse files
feat(mm): clearer naming for main config class hierarchy
1 parent d21707a commit 16570df

File tree

1 file changed

+10
-10
lines changed
  • invokeai/backend/model_manager/configs

1 file changed

+10
-10
lines changed

invokeai/backend/model_manager/configs/main.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ def _has_main_keys(state_dict: dict[str | int, Any]) -> bool:
9898
return False
9999

100100

101-
class Main_Checkpoint_Config_Base(Checkpoint_Config_Base, Main_Config_Base):
101+
class Main_SD_Checkpoint_Config_Base(Checkpoint_Config_Base, Main_Config_Base):
102102
"""Model config for main checkpoint models."""
103103

104104
format: Literal[ModelFormat.Checkpoint] = Field(default=ModelFormat.Checkpoint)
@@ -196,19 +196,19 @@ def _validate_looks_like_main_model(cls, mod: ModelOnDisk) -> None:
196196
raise NotAMatchError("state dict does not look like a main model")
197197

198198

199-
class Main_Checkpoint_SD1_Config(Main_Checkpoint_Config_Base, Config_Base):
199+
class Main_Checkpoint_SD1_Config(Main_SD_Checkpoint_Config_Base, Config_Base):
200200
base: Literal[BaseModelType.StableDiffusion1] = Field(default=BaseModelType.StableDiffusion1)
201201

202202

203-
class Main_Checkpoint_SD2_Config(Main_Checkpoint_Config_Base, Config_Base):
203+
class Main_Checkpoint_SD2_Config(Main_SD_Checkpoint_Config_Base, Config_Base):
204204
base: Literal[BaseModelType.StableDiffusion2] = Field(default=BaseModelType.StableDiffusion2)
205205

206206

207-
class Main_Checkpoint_SDXL_Config(Main_Checkpoint_Config_Base, Config_Base):
207+
class Main_Checkpoint_SDXL_Config(Main_SD_Checkpoint_Config_Base, Config_Base):
208208
base: Literal[BaseModelType.StableDiffusionXL] = Field(default=BaseModelType.StableDiffusionXL)
209209

210210

211-
class Main_Checkpoint_SDXLRefiner_Config(Main_Checkpoint_Config_Base, Config_Base):
211+
class Main_Checkpoint_SDXLRefiner_Config(Main_SD_Checkpoint_Config_Base, Config_Base):
212212
base: Literal[BaseModelType.StableDiffusionXLRefiner] = Field(default=BaseModelType.StableDiffusionXLRefiner)
213213

214214

@@ -425,7 +425,7 @@ def _validate_looks_like_gguf_quantized(cls, mod: ModelOnDisk) -> None:
425425
raise NotAMatchError("state dict does not look like GGUF quantized")
426426

427427

428-
class Main_Diffusers_Config_Base(Diffusers_Config_Base, Main_Config_Base):
428+
class Main_SD_Diffusers_Config_Base(Diffusers_Config_Base, Main_Config_Base):
429429
prediction_type: SchedulerPredictionType = Field()
430430
variant: ModelVariantType = Field()
431431

@@ -525,19 +525,19 @@ def _get_variant_or_raise(cls, mod: ModelOnDisk) -> ModelVariantType:
525525
raise NotAMatchError(f"unrecognized unet in_channels {in_channels} for base '{base}'")
526526

527527

528-
class Main_Diffusers_SD1_Config(Main_Diffusers_Config_Base, Config_Base):
528+
class Main_Diffusers_SD1_Config(Main_SD_Diffusers_Config_Base, Config_Base):
529529
base: Literal[BaseModelType.StableDiffusion1] = Field(BaseModelType.StableDiffusion1)
530530

531531

532-
class Main_Diffusers_SD2_Config(Main_Diffusers_Config_Base, Config_Base):
532+
class Main_Diffusers_SD2_Config(Main_SD_Diffusers_Config_Base, Config_Base):
533533
base: Literal[BaseModelType.StableDiffusion2] = Field(BaseModelType.StableDiffusion2)
534534

535535

536-
class Main_Diffusers_SDXL_Config(Main_Diffusers_Config_Base, Config_Base):
536+
class Main_Diffusers_SDXL_Config(Main_SD_Diffusers_Config_Base, Config_Base):
537537
base: Literal[BaseModelType.StableDiffusionXL] = Field(BaseModelType.StableDiffusionXL)
538538

539539

540-
class Main_Diffusers_SDXLRefiner_Config(Main_Diffusers_Config_Base, Config_Base):
540+
class Main_Diffusers_SDXLRefiner_Config(Main_SD_Diffusers_Config_Base, Config_Base):
541541
base: Literal[BaseModelType.StableDiffusionXLRefiner] = Field(BaseModelType.StableDiffusionXLRefiner)
542542

543543

0 commit comments

Comments
 (0)