File size: 602 Bytes
685e018 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | """Small masked discrete diffusion language model."""
from typing import TYPE_CHECKING
from diffusion_lm.config import ExperimentConfig, ModelConfig, TrainingConfig, load_config
if TYPE_CHECKING:
from diffusion_lm.model import DiffusionTransformer
__all__ = [
"DiffusionTransformer",
"ExperimentConfig",
"ModelConfig",
"TrainingConfig",
"load_config",
]
__version__ = "0.1.0"
def __getattr__(name: str):
if name == "DiffusionTransformer":
from diffusion_lm.model import DiffusionTransformer
return DiffusionTransformer
raise AttributeError(name)
|