diff --git a/setup.py b/setup.py index bdd64fba..5e99df02 100644 --- a/setup.py +++ b/setup.py @@ -17,7 +17,7 @@ "wheel==0.42.0", "setuptools==69.1.0", "cmake==3.28.3", - "transformers[sklearn,sentencepiece, audio, vision]==4.38.1", + "transformers[sklearn,sentencepiece, audio, vision]==4.38.2", "huggingface_hub==0.20.3", "orjson", # vision @@ -30,7 +30,8 @@ # web api "starlette", "uvicorn", - "pandas" + "pandas", + "peft==0.9.0" ] extras = {}