custom_generate/__init__.py

# Custom generate function for fuzzy speculative decoding
from .generate import generate

__all__ = ["generate"]