Skip to content

Commit 69eda20

Browse files
authored
re-enable DS zero3 ci with updated transformers (axolotl-ai-cloud#2533)
1 parent b8c633a commit 69eda20

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

tests/e2e/multigpu/test_llama.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -621,12 +621,6 @@ def test_fsdp_qlora_prequant_packed(self, temp_dir):
621621
temp_dir + "/runs", "train/train_loss", 2.3, "Train Loss is too high"
622622
)
623623

624-
# TODO: remove skip once deepspeed regression is fixed
625-
# see https://github.com/huggingface/transformers/pull/37324
626-
@pytest.mark.skipif(
627-
transformers_version_eq("4.51.0"),
628-
reason="zero3 is not supported with transformers==4.51.0",
629-
)
630624
@pytest.mark.parametrize(
631625
"gradient_accumulation_steps",
632626
[1, 2],

0 commit comments

Comments
 (0)