wassname/alignment-handbook
1
0
Fork 0
mirror of https://github.com/wassname/alignment-handbook.git synced 2026-08-18 11:50:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

CITATION.cff and fix F401 warning (#183)

Browse Source 
* 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>
This commit is contained in:
Kashif Rasul
2024-07-30 10:56:17 +02:00
committed by GitHub
co-authored by Alvaro Bartolome
parent 95dc47218c
commit 98563353d7
4 changed files with 77 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/alignment/__init__.py
+18
View File
@@ -11,3 +11,21 @@ from .model_utils import (
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",
]