scripts/train_bambara.py

"""
Phase 3a: Fine-tune LoRA adapter for Bambara (bam).

Usage:
    python scripts/train_bambara.py
"""
import logging
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent.parent))

from dotenv import load_dotenv

load_dotenv()

logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(name)s — %(message)s")

from src.training.trainer import WhisperLoRATrainer

if __name__ == "__main__":
    trainer = WhisperLoRATrainer(
        base_config_path="configs/base_config.yaml",
        language_config_path="configs/lora_bambara.yaml",
    )
    trainer.setup()
    trainer.train()
    print("\nBambara training complete. Adapter saved to adapters/bambara/")