fixes for dpo and orpo template loading (#1424)

This commit is contained in:
Wing Lian
2024-03-20 11:36:24 -04:00
committed by GitHub
parent dd449c5cd8
commit 7803f0934f
3 changed files with 3 additions and 2 deletions

View File

@@ -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

View File

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

View File

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