mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-06 02:30:30 -08:00
Code Issues Projects Releases Packages Wiki Activity

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

Browse source
This commit is contained in:
AUTOMATIC 2023-01-24 20:24:27 +03:00
parent dac45299dd
commit 28189985e6
3 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
modules/interrogate.py
View file

@ -82,9 +82,16 @@ 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"):
import models.blip
create_fake_fairscale()
import models.blip
files = modelloader.load_models(
model_path=os.path.join(paths.models_path, "BLIP"),