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

Use ungated models for unit tests (#196)

Browse Source 
* Fix unit test

* Fix chat template tests

* Remove deprecated test

* up
This commit is contained in:
lewtun
2024-08-19 12:33:11 +02:00
committed by GitHub
parent 28bf902477
commit a8dcde2cd3
3 changed files with 11 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/alignment/data.py
+1 -1
View File
@@ -32,7 +32,7 @@ def maybe_insert_system_message(messages, tokenizer):
# chat template can be one of two attributes, we check in order
chat_template = tokenizer.chat_template
if chat_template is None:
chat_template = tokenizer.default_chat_template
chat_template = tokenizer.get_chat_template()
# confirm the jinja template refers to a system message before inserting
if "system" in chat_template or "<|im_start|>" in chat_template: