File size: 536 Bytes
1e05592
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
"""
Configuration module for LKAlert training pipeline
"""

from .config import (
    DataConfig,
    ModelConfig,
    SFTTrainingConfig,
    DPOTrainingConfig,
    EvaluationConfig,
    FullConfig,
    ABLATION_CONFIGS,
    create_ablation_config,
    get_debug_config,
    get_fast_config
)

__all__ = [
    'DataConfig',
    'ModelConfig',
    'SFTTrainingConfig',
    'DPOTrainingConfig',
    'EvaluationConfig',
    'FullConfig',
    'ABLATION_CONFIGS',
    'create_ablation_config',
    'get_debug_config',
    'get_fast_config'
]