mirror of
https://github.com/wassname/alignment-handbook.git
synced 2026-07-11 11:56:38 +08:00
Lewis Tunstall
6 lines
273 B
Python
6 lines
273 B
Python
__version__ = "0.2.0.dev0"
|
|
|
|
from .configs import DataArguments, DPOConfig, H4ArgumentParser, ModelArguments, SFTConfig
|
|
from .data import apply_chat_template, get_datasets
|
|
from .model_utils import get_kbit_device_map, get_peft_config, get_quantization_config, get_tokenizer
|