From 48e7b600a8770dd9c8aadb93ef75dd15de2107f9 Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Tue, 19 Sep 2023 18:30:23 +0800
Subject: [PATCH] fix error info

Former-commit-id: 7e8655c8b59c3fdc455e304cf875a6b7fcb69290
---
 src/llmtuner/extras/template.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llmtuner/extras/template.py b/src/llmtuner/extras/template.py
index 3ddd0186..9ae77d5d 100644
--- a/src/llmtuner/extras/template.py
+++ b/src/llmtuner/extras/template.py
@@ -147,7 +147,7 @@ class Template:
             elif isinstance(elem, dict):
                 token_ids = token_ids + [tokenizer.convert_tokens_to_ids(elem.get("token"))]
             else:
-                raise NotImplementedError
+                raise ValueError("Input must be string or dict[str, str], got {}".format(type(elem)))
 
         return token_ids