diff --git a/pyproject.toml b/pyproject.toml index b0caf64..6af3734 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -45,7 +45,7 @@ finetune = [ "transformers>=4.45", "peft>=0.13", "trl>=0.12", - "datasets>=3.0", + "datasets>=5.0.0", "bitsandbytes>=0.44", "accelerate>=1.13.0", ] @@ -74,7 +74,7 @@ all = [ "fastapi>=0.136.3", "uvicorn[standard]>=0.49.0", "starlette>=1.2.1", "redis>=5.0", "httpx[http2]>=0.27", "transformers>=4.45", "peft>=0.13", "trl>=0.12", - "datasets>=3.0", "bitsandbytes>=0.44", "accelerate>=1.13.0", + "datasets>=5.0.0", "bitsandbytes>=0.44", "accelerate>=1.13.0", ] [project.scripts]