remove fairscale requirement, add fake fairscale to make BLIP not complain about it

1 files changed, 9 insertions, 2 deletions

diff --git a/modules/interrogate.py b/modules/interrogate.py
index 236e6983..9f063197 100644
--- a/modules/interrogate.py
+++ b/modules/interrogate.py
@@ -82,9 +82,16 @@ class InterrogateModels:
 

         return self.loaded_categories

 

+    def create_fake_fairscale(self):

+        class FakeFairscale:

+            def checkpoint_wrapper(self):

+                pass

+

+        sys.modules["fairscale.nn.checkpoint.checkpoint_activations"] = FakeFairscale

+

     def load_blip_model(self):

-        with paths.Prioritize("BLIP"):

-            import models.blip

+        create_fake_fairscale()

+        import models.blip

 

         files = modelloader.load_models(

             model_path=os.path.join(paths.models_path, "BLIP"),