diff --git a/finetuner-workflow/finetuner/requirements-precompilable.txt b/finetuner-workflow/finetuner/requirements-precompilable.txt index 8d56fea1..df8759e1 100644 --- a/finetuner-workflow/finetuner/requirements-precompilable.txt +++ b/finetuner-workflow/finetuner/requirements-precompilable.txt @@ -1,3 +1,3 @@ -deepspeed==0.9.2 +deepspeed==0.15.1 flash-attn==1.0.4 einops==0.6.1 \ No newline at end of file