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

feat(mm): add AnyDefaultSettings union

Browse Source
This commit is contained in:
psychedelicious
2024-03-08 15:37:31 +11:00
committed by Brandon
parent 8ce485b036
commit 80c2a4b925
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -22,7 +22,7 @@ Validation errors will raise an InvalidModelConfigException error.
import time import time
from enum import Enum from enum import Enum
from typing import Literal, Optional, Type, Union from typing import Literal, Optional, Type, TypeAlias, Union
import torch import torch
from diffusers.models.modeling_utils import ModelMixin from diffusers.models.modeling_utils import ModelMixin
@ -387,6 +387,7 @@ AnyModelConfig = Annotated[
] ]
AnyModelConfigValidator = TypeAdapter(AnyModelConfig) AnyModelConfigValidator = TypeAdapter(AnyModelConfig)
AnyDefaultSettings: TypeAlias = Union[MainModelDefaultSettings, ControlAdapterDefaultSettings]
class ModelConfigFactory(object): class ModelConfigFactory(object):