35 lines
804 B
Python
35 lines
804 B
Python
__version__ = "1.3.7"
|
|
__author__ = "ViperEkura"
|
|
|
|
from astrai.config import (
|
|
AutoRegressiveLMConfig,
|
|
EncoderConfig,
|
|
TrainConfig,
|
|
)
|
|
from astrai.dataset import DatasetFactory
|
|
from astrai.factory import BaseFactory
|
|
from astrai.inference import (
|
|
GenerationRequest,
|
|
InferenceEngine,
|
|
)
|
|
from astrai.model import AutoModel, AutoRegressiveLM
|
|
from astrai.tokenize import AutoTokenizer
|
|
from astrai.trainer import CallbackFactory, SchedulerFactory, StrategyFactory, Trainer
|
|
|
|
__all__ = [
|
|
"AutoRegressiveLM",
|
|
"AutoRegressiveLMConfig",
|
|
"EncoderConfig",
|
|
"TrainConfig",
|
|
"DatasetFactory",
|
|
"AutoTokenizer",
|
|
"GenerationRequest",
|
|
"InferenceEngine",
|
|
"Trainer",
|
|
"CallbackFactory",
|
|
"StrategyFactory",
|
|
"SchedulerFactory",
|
|
"BaseFactory",
|
|
"AutoModel",
|
|
]
|