Function get_class_in_module

src/diffusers/utils/dynamic_modules_utils.py:185–215

Import a module on the cache directory for modules and extract a class from it.

(class_name, module_path, force_reload=False)

def get_class_in_module(class_name, module_path, force_reload=False):
    """
    Import a module on the cache directory for modules and extract a class from it.
    """
    name = os.path.normpath(module_path)
    if name.endswith(".py"):
        name = name[:-3]
    name = name.replace(os.path.sep, ".")
    module_file: Path = Path(HF_MODULES_CACHE) / module_path

    with _HF_REMOTE_CODE_LOCK:
        if force_reload:
            sys.modules.pop(name, None)
            importlib.invalidate_caches()
        cached_module: ModuleType | None = sys.modules.get(name)
        module_spec = importlib.util.spec_from_file_location(name, location=module_file)

        module: ModuleType
        if cached_module is None:
            module = importlib.util.module_from_spec(module_spec)
            # insert it into sys.modules before any loading begins
            sys.modules[name] = module
        else:
            module = cached_module

        module_spec.loader.exec_module(module)

    if class_name is None:
        return find_pipeline_class(module)

    return getattr(module, class_name)


def find_pipeline_class(loaded_module):