fixes for dpo and orpo template loading (#1424)

axolotl-ai-cloud · Mar 20, 2024 · 7803f09 · 7803f09
1 parent dd449c5
commit 7803f09
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -34,7 +34,7 @@ jobs:
       fail-fast: false
       matrix:
         python_version: ["3.10", "3.11"]
-    timeout-minutes: 10
+    timeout-minutes: 20
 
     steps:
       - name: Check out repository code

diff --git a/src/axolotl/prompt_strategies/dpo/__init__.py b/src/axolotl/prompt_strategies/dpo/__init__.py
@@ -5,4 +5,4 @@
 
 from ..base import load as load_base
 
-load = partial(load_base, module="axolotl.prompt_strategies.dpo")
+load = partial(load_base, module_base="axolotl.prompt_strategies.dpo")
diff --git a/src/axolotl/prompt_strategies/orpo/chat_template.py b/src/axolotl/prompt_strategies/orpo/chat_template.py
@@ -36,6 +36,7 @@ def load(
             chat_template = chat_templates(chat_template)
         except ValueError:
             pass
+    tokenizer.chat_template = chat_template
 
     return ORPOTokenizingStrategy(
         ORPOPrompter(chat_template, tokenizer),
Original file line number	Diff line number	Diff line change
Expand Up		@@ -5,4 +5,4 @@

		from ..base import load as load_base

		load = partial(load_base, module="axolotl.prompt_strategies.dpo")
		load = partial(load_base, module_base="axolotl.prompt_strategies.dpo")