@@ -98,7 +98,7 @@ def _has_main_keys(state_dict: dict[str | int, Any]) -> bool:
98
98
return False
99
99
100
100
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 ):
102
102
"""Model config for main checkpoint models."""
103
103
104
104
format : Literal [ModelFormat .Checkpoint ] = Field (default = ModelFormat .Checkpoint )
@@ -196,19 +196,19 @@ def _validate_looks_like_main_model(cls, mod: ModelOnDisk) -> None:
196
196
raise NotAMatchError ("state dict does not look like a main model" )
197
197
198
198
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 ):
200
200
base : Literal [BaseModelType .StableDiffusion1 ] = Field (default = BaseModelType .StableDiffusion1 )
201
201
202
202
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 ):
204
204
base : Literal [BaseModelType .StableDiffusion2 ] = Field (default = BaseModelType .StableDiffusion2 )
205
205
206
206
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 ):
208
208
base : Literal [BaseModelType .StableDiffusionXL ] = Field (default = BaseModelType .StableDiffusionXL )
209
209
210
210
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 ):
212
212
base : Literal [BaseModelType .StableDiffusionXLRefiner ] = Field (default = BaseModelType .StableDiffusionXLRefiner )
213
213
214
214
@@ -425,7 +425,7 @@ def _validate_looks_like_gguf_quantized(cls, mod: ModelOnDisk) -> None:
425
425
raise NotAMatchError ("state dict does not look like GGUF quantized" )
426
426
427
427
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 ):
429
429
prediction_type : SchedulerPredictionType = Field ()
430
430
variant : ModelVariantType = Field ()
431
431
@@ -525,19 +525,19 @@ def _get_variant_or_raise(cls, mod: ModelOnDisk) -> ModelVariantType:
525
525
raise NotAMatchError (f"unrecognized unet in_channels { in_channels } for base '{ base } '" )
526
526
527
527
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 ):
529
529
base : Literal [BaseModelType .StableDiffusion1 ] = Field (BaseModelType .StableDiffusion1 )
530
530
531
531
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 ):
533
533
base : Literal [BaseModelType .StableDiffusion2 ] = Field (BaseModelType .StableDiffusion2 )
534
534
535
535
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 ):
537
537
base : Literal [BaseModelType .StableDiffusionXL ] = Field (BaseModelType .StableDiffusionXL )
538
538
539
539
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 ):
541
541
base : Literal [BaseModelType .StableDiffusionXLRefiner ] = Field (BaseModelType .StableDiffusionXLRefiner )
542
542
543
543
0 commit comments