diff --git a/pyproject.toml b/pyproject.toml index b0caf64..0eb38e9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -43,7 +43,7 @@ gateway = ["fastapi>=0.136.3", "uvicorn[standard]>=0.49.0", "starlette>=1.2.1", # Fine-tuning pipeline (llmstack finetune) finetune = [ "transformers>=4.45", - "peft>=0.13", + "peft>=0.19.1", "trl>=0.12", "datasets>=3.0", "bitsandbytes>=0.44", @@ -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.19.1", "trl>=0.12", "datasets>=3.0", "bitsandbytes>=0.44", "accelerate>=1.13.0", ]