diff --git a/pgml-extension/requirements.txt b/pgml-extension/requirements.txt index 6dd8b4690..235c726ef 100644 --- a/pgml-extension/requirements.txt +++ b/pgml-extension/requirements.txt @@ -28,3 +28,4 @@ einops==0.6.1 pynvml==11.5.0 transformers-stream-generator==0.0.4 optimum==1.13.2 +peft==0.6.2