Commit a04c9551 authored by AUTOMATIC1111's avatar AUTOMATIC1111

fix importlib.machinery issue on github's autotests #yolo

parent 5cf623c5
...@@ -227,13 +227,14 @@ def run_extensions_installers(settings_file): ...@@ -227,13 +227,14 @@ def run_extensions_installers(settings_file):
def mute_sdxl_imports(): def mute_sdxl_imports():
"""create fake modules that SDXL wants to import but doesn't actually use for our purposes""" """create fake modules that SDXL wants to import but doesn't actually use for our purposes"""
import importlib class Dummy:
pass
module = importlib.util.module_from_spec(importlib.machinery.ModuleSpec('taming.modules.losses.lpips', None)) module = Dummy()
module.LPIPS = None module.LPIPS = None
sys.modules['taming.modules.losses.lpips'] = module sys.modules['taming.modules.losses.lpips'] = module
module = importlib.util.module_from_spec(importlib.machinery.ModuleSpec('sgm.data', None)) module = Dummy()
module.StableDataModuleFromConfig = None module.StableDataModuleFromConfig = None
sys.modules['sgm.data'] = module sys.modules['sgm.data'] = module
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment