Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(mm): remove ztsnr_training field on _MainConfig

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/app/api/routers/model_manager.py
View File

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

2
invokeai/app/services/model_install/model_install_base.py
View File

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

1
invokeai/backend/model_manager/config.py
View File

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