28 lines
632 B
Python
28 lines
632 B
Python
|
"""
|
||
|
|
Training utilities for VoxCPM fine-tuning.
|
||
|
|
|
||
|
|
This package mirrors the training mechanics used in the minicpm-audio
|
||
|
|
tooling while relying solely on local audio-text datasets managed via
|
||
|
|
the HuggingFace ``datasets`` library.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from .accelerator import Accelerator
|
||
|
|
from .tracker import TrainingTracker
|
||
|
|
from .data import (
|
||
|
|
load_audio_text_datasets,
|
||
|
|
HFVoxCPMDataset,
|
||
|
|
build_dataloader,
|
||
|
|
BatchProcessor,
|
||
|
|
)
|
||
|
|
from .state import TrainingState
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"Accelerator",
|
||
|
|
"TrainingTracker",
|
||
|
|
"HFVoxCPMDataset",
|
||
|
|
"BatchProcessor",
|
||
|
|
"TrainingState",
|
||
|
|
"load_audio_text_datasets",
|
||
|
|
"build_dataloader",
|
||
|
|
]
|