tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fixes for dpo and orpo template loading (#1424)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/tests.yml vendored
View File

@@ -34,7 +34,7 @@ jobs:
fail-fast: false fail-fast: false
matrix: matrix:
python_version: ["3.10", "3.11"] python_version: ["3.10", "3.11"]
timeout-minutes: 10 timeout-minutes: 20
steps: steps:
- name: Check out repository code - name: Check out repository code

2
src/axolotl/prompt_strategies/dpo/__init__.py
View File

@@ -5,4 +5,4 @@ from functools import partial
from ..base import load as load_base 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")

1
src/axolotl/prompt_strategies/orpo/chat_template.py
View File

@@ -36,6 +36,7 @@ def load(
chat_template = chat_templates(chat_template) chat_template = chat_templates(chat_template)
except ValueError: except ValueError:
pass pass
tokenizer.chat_template = chat_template
return ORPOTokenizingStrategy( return ORPOTokenizingStrategy(
ORPOPrompter(chat_template, tokenizer), ORPOPrompter(chat_template, tokenizer),