diff --git a/invokeai/app/services/model_manager_service.py b/invokeai/app/services/model_manager_service.py index fd24aca81b..8384a97435 100644 --- a/invokeai/app/services/model_manager_service.py +++ b/invokeai/app/services/model_manager_service.py @@ -3,7 +3,7 @@ from abc import ABC, abstractmethod from pathlib import Path -from typing import Union, Callable, types +from typing import Union, Callable, List, Tuple, types from dataclasses import dataclass from invokeai.backend.model_management.model_manager import ( @@ -67,7 +67,7 @@ class ModelManagerServiceBase(ABC): pass @abstractmethod - def default_model(self) -> Union[Tuple(str, SDModelType),None], + def default_model(self) -> Union[Tuple(str, SDModelType),None]: """ Returns the name and typeof the default model, or None if none is defined. @@ -352,7 +352,7 @@ class ModelManagerService(ModelManagerServiceBase): model_name, model_type) - def default_model(self) -> Union[Tuple(str, SDModelType),None], + def default_model(self) -> Union[Tuple(str, SDModelType),None]: """ Returns the name of the default model, or None if none is defined.