diff --git a/turbo_alignment/dataset/chat/chat.py b/turbo_alignment/dataset/chat/chat.py index c96af58f..316a7502 100755 --- a/turbo_alignment/dataset/chat/chat.py +++ b/turbo_alignment/dataset/chat/chat.py @@ -206,6 +206,7 @@ def _truncate_and_merge( truncated_conversation_messages = [conversation.messages[0]] + truncated_conversation_messages truncated_tokenized_replicas = [truncated_tokenized_replicas[0]] + truncated_tokenized_replicas + for ind, (message, tokenized_replica) in enumerate( zip( truncated_conversation_messages,