refactor(mm): remove ztsnr_training field on _MainConfig

This is used to determine the CFG Rescale Multiplier setting. We'll handle this in the UI as a default setting.
This commit is contained in:
psychedelicious 2024-03-01 13:03:32 +11:00
parent dd31bc4586
commit b880a31039
3 changed files with 1 additions and 3 deletions

View File

@ -79,7 +79,6 @@ example_model_config = {
"prediction_type": "epsilon",
"repo_variant": "fp16",
"upcast_attention": False,
"ztsnr_training": False,
}
example_model_input = {

View File

@ -392,7 +392,7 @@ class ModelInstallServiceBase(ABC):
will override corresponding autoassigned probe fields in the
model's config record. Use it to override
`name`, `description`, `base_type`, `model_type`, `format`,
`prediction_type`, `image_size`, and/or `ztsnr_training`.
`prediction_type`, and/or `image_size`.
This will download the model located at `source`,
probe it, and install it into the models directory.

View File

@ -263,7 +263,6 @@ class _MainConfig(ModelConfigBase):
variant: ModelVariantType = ModelVariantType.Normal
prediction_type: SchedulerPredictionType = SchedulerPredictionType.Epsilon
upcast_attention: bool = False
ztsnr_training: bool = False
class MainCheckpointConfig(_CheckpointConfig, _MainConfig):