Commit 28189985 authored by AUTOMATIC's avatar AUTOMATIC

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

parent dac45299
...@@ -82,9 +82,16 @@ class InterrogateModels: ...@@ -82,9 +82,16 @@ class InterrogateModels:
return self.loaded_categories 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): def load_blip_model(self):
with paths.Prioritize("BLIP"): create_fake_fairscale()
import models.blip import models.blip
files = modelloader.load_models( files = modelloader.load_models(
model_path=os.path.join(paths.models_path, "BLIP"), model_path=os.path.join(paths.models_path, "BLIP"),
......
...@@ -14,7 +14,6 @@ scikit-image==0.19.2 ...@@ -14,7 +14,6 @@ scikit-image==0.19.2
fonts fonts
font-roboto font-roboto
timm==0.6.7 timm==0.6.7
fairscale==0.4.9
piexif==1.1.3 piexif==1.1.3
einops==0.4.1 einops==0.4.1
jsonmerge==1.8.0 jsonmerge==1.8.0
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment