diff --git a/__init__.py b/__init__.py deleted file mode 100644 index debe74e..0000000 --- a/__init__.py +++ /dev/null @@ -1,19 +0,0 @@ -""" -RL Fine-tuning Library - -A modular library for reinforcement learning fine-tuning of language models. -""" - -__version__ = "0.0.1" - -from .core.config import ExperimentConfig, ModelConfig, DataConfig, TrainingConfig, WandbConfig -from .trainers.dpo_trainer import DPOTrainer - -__all__ = [ - "ExperimentConfig", - "ModelConfig", - "DataConfig", - "TrainingConfig", - "WandbConfig", - "DPOTrainer", -]