mirror of
https://github.com/OpenBMB/VoxCPM
synced 2025-12-12 03:48:12 +00:00
29 lines
633 B
Python
29 lines
633 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",
|
|
]
|
|
|