diff --git a/pyproject.toml b/pyproject.toml index 24ce9459..ecf3b624 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -28,7 +28,7 @@ dependencies = [ "torch>=2.0.1", "tqdm>=4.65.0", "transformers>=4.32.0", - "peft==0.6.0", + "peft==0.7.0,<0.8.0", ] [tool.setuptools.packages.find]