Commit 28189985 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

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

parent dac45299

modules/interrogate.py

+9 −2
Original line number Diff line number Diff line
@@ -82,8 +82,15 @@ class InterrogateModels: 


        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):

        with paths.Prioritize("BLIP"):

        create_fake_fairscale()

        import models.blip



        files = modelloader.load_models(

requirements.txt

+0 −1
Original line number Diff line number Diff line 
blendmodes

accelerate

basicsr

fairscale==0.4.4

fonts

font-roboto

gfpgan

requirements_versions.txt

+0 −1
Original line number Diff line number Diff line
@@ -14,7 +14,6 @@ scikit-image==0.19.2 
fonts

font-roboto

timm==0.6.7

fairscale==0.4.9

piexif==1.1.3

einops==0.4.1

jsonmerge==1.8.0

CRA Git | Maintained and supported by SUSTech CRA and CCSE