diff --git a/requirements.txt b/requirements.txt index 9c56638a3..e1f1b10a5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,11 +13,11 @@ packaging==23.2 huggingface_hub>=0.33.0 peft>=0.17.1 tokenizers>=0.21.1 -transformers==4.57.0 +transformers==4.57.1 accelerate==1.10.1 datasets==4.0.0 deepspeed>=0.17.0 -trl==0.23.0 +trl==0.23.1 hf_xet==1.1.5 kernels==0.9.0 trackio diff --git a/tests/e2e/multigpu/test_llama.py b/tests/e2e/multigpu/test_llama.py index ffdbad942..3383e71d1 100644 --- a/tests/e2e/multigpu/test_llama.py +++ b/tests/e2e/multigpu/test_llama.py @@ -546,7 +546,6 @@ class TestMultiGPULlama: temp_dir + "/runs", "train/train_loss", 2.1, "Train Loss (%s) is too high" ) - @pytest.mark.skip("regression failure from v4.57.0") def test_fsdp_qlora_prequant_packed(self, temp_dir): cfg = DictDefault( {