From 04121f574d4e735473a388f1e526a67d24599620 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 2 Dec 2024 23:59:00 -0500 Subject: [PATCH] fix messages field for test dataset --- tests/e2e/test_relora_llama.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/e2e/test_relora_llama.py b/tests/e2e/test_relora_llama.py index 0baf2c9bae..56c2204677 100644 --- a/tests/e2e/test_relora_llama.py +++ b/tests/e2e/test_relora_llama.py @@ -57,7 +57,7 @@ def test_relora(self, temp_dir): "path": "mlabonne/FineTome-100k", "type": "chat_template", "split": "train[:10%]", - "field_messages": "conversation", + "field_messages": "conversations", "message_field_role": "from", "message_field_content": "value", },