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

@ -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):