diffusion_llm/inference/__init__.py

"""Inference module for AAM Diffusion LLM."""

from diffusion_llm.inference.generator import AamGenerator

__all__ = ["AamGenerator"]