remove fairscale requirement, add fake fairscale to make BLIP not complain about it mk2
This commit is contained in:
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…
Reference in New Issue
Block a user