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

test

Browse Source
This commit is contained in:
sunny
2024-11-05 12:35:36 -05:00
parent bcb1205e39
commit d9b65f69fb
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/e2e/test_dpo.py
View File

@@ -27,7 +27,7 @@ class TestDPOLlamaLora(unittest.TestCase):
"""
@with_temp_dir
def test_dpo_lora(self, temp_dir):
def test_dpo_nll_lora(self, temp_dir):
# pylint: disable=duplicate-code
cfg = DictDefault(
{
@@ -42,6 +42,7 @@ class TestDPOLlamaLora(unittest.TestCase):
"lora_target_linear": True,
"special_tokens": {},
"rl": "dpo",
"rpo_alpha": 0.5,
"datasets": [
{
"path": "arcee-ai/distilabel-intel-orca-dpo-pairs-binarized",
@@ -71,7 +72,7 @@ class TestDPOLlamaLora(unittest.TestCase):
assert (Path(temp_dir) / "checkpoint-20/adapter_model.safetensors").exists()
@with_temp_dir
def test_dpo_nll_lora(self, temp_dir):
def test_dpo_lora(self, temp_dir):
# pylint: disable=duplicate-code
cfg = DictDefault(
{
@@ -86,7 +87,6 @@ class TestDPOLlamaLora(unittest.TestCase):
"lora_target_linear": True,
"special_tokens": {},
"rl": "dpo",
"rpo_alpha": 0.5,
"datasets": [
{
"path": "arcee-ai/distilabel-intel-orca-dpo-pairs-binarized",