From 5c1cb9263f980641007088a37360fcab01761d37 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Tue, 24 Jan 2023 00:24:17 +0300
Subject: fix BLIP failing to import depending on configuration

---
 modules/interrogate.py |  3 ++-
 modules/paths.py       | 14 ++++++++++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/modules/interrogate.py b/modules/interrogate.py
index c252b148..236e6983 100644
--- a/modules/interrogate.py
+++ b/modules/interrogate.py
@@ -83,7 +83,8 @@ class InterrogateModels:
         return self.loaded_categories
 
     def load_blip_model(self):
-        import models.blip
+        with paths.Prioritize("BLIP"):
+            import models.blip
 
         files = modelloader.load_models(
             model_path=os.path.join(paths.models_path, "BLIP"),
diff --git a/modules/paths.py b/modules/paths.py
index 4dd03a35..20b3e4d8 100644
--- a/modules/paths.py
+++ b/modules/paths.py
@@ -38,3 +38,17 @@ for d, must_exist, what, options in path_dirs:
         else:
             sys.path.append(d)
         paths[what] = d
+
+
+class Prioritize:
+    def __init__(self, name):
+        self.name = name
+        self.path = None
+
+    def __enter__(self):
+        self.path = sys.path.copy()
+        sys.path = [paths[self.name]] + sys.path
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        sys.path = self.path
+        self.path = None
-- 
cgit v1.2.1