remove fairscale requirement, add fake fairscale to make BLIP not complain about it
This commit is contained in:
parent
dac45299dd
commit
28189985e6
3 changed files with 9 additions and 4 deletions
|
@ -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"),
|
||||||
|
|
|
@ -1,7 +1,6 @@
|
||||||
blendmodes
|
blendmodes
|
||||||
accelerate
|
accelerate
|
||||||
basicsr
|
basicsr
|
||||||
fairscale==0.4.4
|
|
||||||
fonts
|
fonts
|
||||||
font-roboto
|
font-roboto
|
||||||
gfpgan
|
gfpgan
|
||||||
|
|
|
@ -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
|
||||||
|
|
Loading…
Reference in a new issue