[Bug] fix openai_map_fn bugs (#885)

fix openai_map_fn bugs
InternLM · Aug 9, 2024 · 7dd779b · 7dd779b
1 parent 01640b0
commit 7dd779b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/xtuner/dataset/map_fns/dataset_map_fns/openai_map_fn.py b/xtuner/dataset/map_fns/dataset_map_fns/openai_map_fn.py
@@ -32,7 +32,8 @@ def openai_map_fn(example):
         elif msg['role'] == 'user':
             input += msg['content']
         elif msg['role'] == 'assistant':
-            output_with_loss = msg.get('loss', True)
+            output_with_loss = msg.get('loss', 'True')
+            output_with_loss = str(output_with_loss)
             output_with_loss = output_with_loss.lower() == 'true'
             conversation.append({
                 'system': system,