Revert some files

This commit is contained in:
Billy
2025-03-26 13:18:50 +11:00
parent 00b11ef795
commit a6b94e8ca4
3 changed files with 8 additions and 1 deletions

View File

@@ -1,6 +1,8 @@
from .config import set_layer_config
from .helpers import load_checkpoint
from .gen_efficientnet import *
from .mobilenetv3 import *
def create_model(

View File

@@ -32,10 +32,14 @@ from invokeai.app.services.model_install.model_install_common import (
URLModelSource,
)
from invokeai.app.services.model_records import ModelRecordChanges, UnknownModelException
from invokeai.backend.model_manager import BaseModelType, ModelFormat, ModelRepoVariant, ModelType
from invokeai.backend.model_manager.config import (
BaseModelType,
InvalidModelConfigException,
ModelFormat,
ModelRepoVariant,
ModelType,
)
from tests.backend.model_manager.model_manager_fixtures import * # noqa F403
from tests.test_nodes import TestEventService
OS = platform.uname().system

View File

@@ -12,6 +12,7 @@ from invokeai.app.services.shared.invocation_context import (
build_invocation_context,
)
from invokeai.backend.model_manager.load.load_base import LoadedModelWithoutConfig
from tests.backend.model_manager.model_manager_fixtures import * # noqa F403
@pytest.fixture()