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

Add skeleton

Browse Source
This commit is contained in:
Lewis Tunstall
2023-11-08 13:21:57 +00:00
parent b9d9aa0a29
commit 967eab4cfb
13 changed files with 971 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/alignment/__init__.py
+4
View File
@@ -1 +1,5 @@
__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