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

master
AUTOMATIC 3 years ago
parent 28189985e6
commit 5228ec8bda

@ -90,7 +90,7 @@ class InterrogateModels:
sys.modules["fairscale.nn.checkpoint.checkpoint_activations"] = FakeFairscale sys.modules["fairscale.nn.checkpoint.checkpoint_activations"] = FakeFairscale
def load_blip_model(self): def load_blip_model(self):
create_fake_fairscale() self.create_fake_fairscale()
import models.blip import models.blip
files = modelloader.load_models( files = modelloader.load_models(

Loading…
Cancel
Save