VLAlert / training /configs /__init__.py
AsianPlayer's picture
Add VLAlert code
1e05592 verified
Raw
History Blame Contribute Delete
536 Bytes
"""
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'
]