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

This commit is contained in:
AUTOMATIC 2023-01-24 20:30:43 +03:00
parent 28189985e6
commit 5228ec8bda
1 changed files with 1 additions and 1 deletions

View File

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