Commit 5228ec8b authored by AUTOMATIC's avatar AUTOMATIC
Browse files

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

parent 28189985

modules/interrogate.py

+1 −1
Original line number Diff line number Diff line
@@ -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(

CRA Git | Maintained and supported by SUSTech CRA and CCSE