fix for inference keep end

turbo-llm · Nov 13, 2024 · 5c4c742 · 5c4c742
1 parent a042f07
commit 5c4c742
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,