mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-23 06:12:50 +08:00
fix test
Former-commit-id: c5e1736470c4c54de5a89ea1c58154b7fed2405d
This commit is contained in:
parent
d5ea05cfff
commit
60d770e4b1
@ -117,7 +117,7 @@ def test_encode_multiturn(use_fast: bool):
|
||||
def test_jinja_template(use_fast: bool):
|
||||
tokenizer = AutoTokenizer.from_pretrained(TINY_LLAMA, use_fast=use_fast)
|
||||
ref_tokenizer = AutoTokenizer.from_pretrained(TINY_LLAMA, use_fast=use_fast)
|
||||
get_template_and_fix_tokenizer(tokenizer, name="llama3")
|
||||
get_template_and_fix_tokenizer(tokenizer, DataArguments(template="llama3"))
|
||||
assert tokenizer.chat_template != ref_tokenizer.chat_template
|
||||
assert tokenizer.apply_chat_template(MESSAGES) == ref_tokenizer.apply_chat_template(MESSAGES)
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user