diff --git a/py/AILab_RMBG.py b/py/AILab_RMBG.py index 06c5a63..3db7f26 100644 --- a/py/AILab_RMBG.py +++ b/py/AILab_RMBG.py @@ -187,6 +187,7 @@ def load_model(self, model_name): module_name = f"custom_birefnet_model_{hash(birefnet_path)}" module = types.ModuleType(module_name) + module.__file__ = birefnet_path sys.modules[module_name] = module exec(birefnet_content, module.__dict__)