From a6b94e8ca46c1ad4f80cc6df85624bea489637f7 Mon Sep 17 00:00:00 2001
From: Billy <billy@invoke.ai>
Date: Wed, 26 Mar 2025 13:18:50 +1100
Subject: [PATCH] Revert some files

---
 .../submodules/efficientnet_repo/geffnet/model_factory.py   | 2 ++
 tests/app/services/model_install/test_model_install.py      | 6 +++++-
 tests/app/services/model_load/test_load_api.py              | 1 +
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/invokeai/backend/image_util/normal_bae/nets/submodules/efficientnet_repo/geffnet/model_factory.py b/invokeai/backend/image_util/normal_bae/nets/submodules/efficientnet_repo/geffnet/model_factory.py
index 9830fb1213..4d46ea8bae 100644
--- a/invokeai/backend/image_util/normal_bae/nets/submodules/efficientnet_repo/geffnet/model_factory.py
+++ b/invokeai/backend/image_util/normal_bae/nets/submodules/efficientnet_repo/geffnet/model_factory.py
@@ -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(
diff --git a/tests/app/services/model_install/test_model_install.py b/tests/app/services/model_install/test_model_install.py
index 6ab5862638..88aaf391ab 100644
--- a/tests/app/services/model_install/test_model_install.py
+++ b/tests/app/services/model_install/test_model_install.py
@@ -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
diff --git a/tests/app/services/model_load/test_load_api.py b/tests/app/services/model_load/test_load_api.py
index 6b810081b5..c0760cd3ca 100644
--- a/tests/app/services/model_load/test_load_api.py
+++ b/tests/app/services/model_load/test_load_api.py
@@ -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()