fixes for dpo and orpo template loading (#1424)

2024-03-20 11:36:24 -04:00
parent dd449c5cd8
commit 7803f0934f
3 changed files with 3 additions and 2 deletions
--- a/src/axolotl/prompt_strategies/dpo/init.py
+++ b/src/axolotl/prompt_strategies/dpo/init.py
@@ -5,4 +5,4 @@ from functools import partial

 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")
--- 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),