File size: 220 Bytes
9e30969 | 1 2 3 4 5 6 7 8 9 10 11 | """inference package — text generation and sampling."""
from inference.generate import generate, sample_logits
from inference.sampler import Sampler
__all__ = [
"generate",
"sample_logits",
"Sampler",
]
|