diff --git a/pyproject.toml b/pyproject.toml index 73c9743..0837fb3 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -21,7 +21,7 @@ func-timeout = "4.3.5" datasets = ">=2.14.6" trl = ">=0.5.0" pydantic = "1.10.11" -deepspeed = "0.10.2" +deepspeed = "0.15.1" gradio = ">=3.36.0" uvicorn-http2 = "^0.0.0" uvicorn = "^0.24.0.post1"