add tokenize func

2 weeks ago · c606d1101c
parent b6ec337f3d
commit c606d1101c
1 changed files with 1 additions and 1 deletions
--- a/applications/ColossalChat/coati/dataset/init.py
+++ b/applications/ColossalChat/coati/dataset/init.py
@ -23,5 +23,5 @@ __all__ = [
    "tokenize_kto",
    "setup_conversation_template",
    "Conversation",
-    "tokenize_process_reward"
+    "tokenize_process_reward",
 ]