mirror of
https://github.com/wassname/alignment-handbook.git
synced 2026-06-27 16:14:07 +08:00
Kashif Rasul
98563353d7
* fix F401 warning * add CITATION.cff * update version in CITATION * update title * fix label * Update src/alignment/__init__.py Co-authored-by: Alvaro Bartolome <36760800+alvarobartt@users.noreply.github.com> * make style * add Alvaro Bartolome * update version in readme --------- Co-authored-by: Alvaro Bartolome <36760800+alvarobartt@users.noreply.github.com>
32 lines
730 B
Python
32 lines
730 B
Python
__version__ = "0.3.0.dev0"
|
|
|
|
from .configs import DataArguments, DPOConfig, H4ArgumentParser, ModelArguments, SFTConfig
|
|
from .data import apply_chat_template, get_datasets
|
|
from .decontaminate import decontaminate_humaneval
|
|
from .model_utils import (
|
|
get_checkpoint,
|
|
get_kbit_device_map,
|
|
get_peft_config,
|
|
get_quantization_config,
|
|
get_tokenizer,
|
|
is_adapter_model,
|
|
)
|
|
|
|
|
|
__all__ = [
|
|
"DataArguments",
|
|
"DPOConfig",
|
|
"H4ArgumentParser",
|
|
"ModelArguments",
|
|
"SFTConfig",
|
|
"apply_chat_template",
|
|
"get_datasets",
|
|
"decontaminate_humaneval",
|
|
"get_checkpoint",
|
|
"get_kbit_device_map",
|
|
"get_peft_config",
|
|
"get_quantization_config",
|
|
"get_tokenizer",
|
|
"is_adapter_model",
|
|
]
|