diff --git a/pyproject.toml b/pyproject.toml index b0caf64..b2e4749 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -44,7 +44,7 @@ gateway = ["fastapi>=0.136.3", "uvicorn[standard]>=0.49.0", "starlette>=1.2.1", finetune = [ "transformers>=4.45", "peft>=0.13", - "trl>=0.12", + "trl>=1.6.0", "datasets>=3.0", "bitsandbytes>=0.44", "accelerate>=1.13.0", @@ -73,7 +73,7 @@ all = [ "numpy>=1.24", "pymupdf>=1.27.2.3", "python-docx>=1.2.0", "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", + "transformers>=4.45", "peft>=0.13", "trl>=1.6.0", "datasets>=3.0", "bitsandbytes>=0.44", "accelerate>=1.13.0", ]