ai_exec / src /training /__init__.py
Chaitanya-aitf's picture
Upload 38 files
45ee481 verified
"""
Training module for AI Executive System.
This module handles:
- Dataset preparation for training
- QLoRA fine-tuning
- Standard LoRA training
- Adapter merging and export
"""
from .prepare_dataset import prepare_dataset
from .train_qlora import train_qlora
from .train_lora import train_lora
from .merge_adapter import merge_adapter
__all__ = [
"prepare_dataset",
"train_qlora",
"train_lora",
"merge_adapter",
]