don't fail fatally when turns are improper

axolotl-ai-cloud · Sep 26, 2023 · 654bbcc · 654bbcc
1 parent 4483f1b
commit 654bbcc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/axolotl/prompters.py b/src/axolotl/prompters.py
@@ -276,7 +276,8 @@ def build_prompt(self, source) -> Generator[str, None, None]:
         conv.messages = []
         for j, sentence in enumerate(source):
             role = roles[sentence["from"]]
-            assert role == conv.roles[j % 2], SHAREGPT_ASSERTION_FAILED_ROLE
+            if role != conv.roles[j % 2]:
+                LOG.warning(f"{SHAREGPT_ASSERTION_FAILED_ROLE}: {sentence}")
             conv.append_message(role, sentence["value"])
 
         for part in conv.get_turns():