File size: 435 Bytes
45ee481
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
"""
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",
]