diff options
| author | Xe Iaso <me@xeiaso.net> | 2024-06-29 12:41:33 -0400 |
|---|---|---|
| committer | Xe Iaso <me@xeiaso.net> | 2024-06-29 12:41:33 -0400 |
| commit | f5ea7532f9634089e53d6902fcad432b5cc1c726 (patch) | |
| tree | 946179e159f353b89d37a6f72e740690e2a87be9 | |
| parent | c684932e80d76d6df2d4ba1201ca26ff122dc57b (diff) | |
| download | x-f5ea7532f9634089e53d6902fcad432b5cc1c726.tar.xz x-f5ea7532f9634089e53d6902fcad432b5cc1c726.zip | |
llm: fix test
Signed-off-by: Xe Iaso <me@xeiaso.net>
| -rw-r--r-- | llm/chatml_test.go | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/llm/chatml_test.go b/llm/chatml_test.go index 2a1ff80..3352b71 100644 --- a/llm/chatml_test.go +++ b/llm/chatml_test.go @@ -19,7 +19,8 @@ func TestChatML(t *testing.T) { } expected := `<|im_start|>user -hello<|im_end|> +hello +<|im_end|> <|im_start|>assistant` if strings.TrimSpace(session.ChatML()) != strings.TrimSpace(expected) { |