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

fsdp2 w evals fixed upstream

Browse Source
This commit is contained in:
Wing Lian
2025-08-11 16:26:42 -04:00
parent 8efdc59796
commit 9c0fa60220
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/monkeypatch/test_trainer_loss_calc.py
View File

@@ -3,7 +3,6 @@
import unittest import unittest
from axolotl.monkeypatch.transformers.trainer_loss_calc import ( from axolotl.monkeypatch.transformers.trainer_loss_calc import (
check_evaluation_loop_is_fsdp2_patchable,
check_evaluation_loop_is_patchable, check_evaluation_loop_is_patchable,
check_maybe_log_save_evaluate_is_patchable, check_maybe_log_save_evaluate_is_patchable,
) )
@@ -20,7 +19,6 @@ class TestTrainerLossCalc(unittest.TestCase):
the patched code changes upstream. the patched code changes upstream.
""" """
assert check_evaluation_loop_is_patchable() assert check_evaluation_loop_is_patchable()
assert check_evaluation_loop_is_fsdp2_patchable()
assert check_maybe_log_save_evaluate_is_patchable() assert check_maybe_log_save_evaluate_is_patchable()