Blame for dssd/__init__.py

"""DSSD: exact self-speculative decoding with pruned exit heads and context copying."""

__version__ = "0.1.0"


def __getattr__(name):
    # lazy: `import dssd` must not import transformers (the engine modules do not need it)
    if name in ("DSSD", "Generation", "StreamEvent"):
        from . import api
        return getattr(api, name)
    raise AttributeError(f"module 'dssd' has no attribute {name!r}")