From 41d6ca0058f51bc0ca706e81ba417515f01501c2 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 16 Nov 2023 15:23:45 +0100 Subject: [PATCH] correct peft --- setup.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 68cad46fbe..5f908392c2 100644 --- a/setup.py +++ b/setup.py @@ -112,7 +112,7 @@ _deps = [ "numpy", "omegaconf", "parameterized", - "peft==~0.6", + "peft<=0.6.2", "protobuf>=3.20.3,<4", "pytest", "pytest-timeout", @@ -202,7 +202,6 @@ extras = {} extras["quality"] = deps_list("urllib3", "black", "isort", "ruff", "hf-doc-builder") extras["docs"] = deps_list("hf-doc-builder") extras["training"] = deps_list("accelerate", "datasets", "protobuf", "tensorboard", "Jinja2") -extras["peft"] = deps_list("peft") extras["test"] = deps_list( "compel", "datasets",