From 6b3f64783104b9547ff7aadd179daa400e6d970f Mon Sep 17 00:00:00 2001 From: Montana Low Date: Wed, 13 Dec 2023 13:00:56 -0800 Subject: [PATCH] add missing linux requirements --- pgml-extension/requirements.txt | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pgml-extension/requirements.txt b/pgml-extension/requirements.txt index d8471b5cd..ddc7229b6 100644 --- a/pgml-extension/requirements.txt +++ b/pgml-extension/requirements.txt @@ -4,6 +4,7 @@ aiosignal==1.3.1 annotated-types==0.6.0 anyio==4.1.0 attrs==23.1.0 +auto-gptq==0.5.1; sys_platform == 'linux' bitsandbytes==0.41.3.post2 catboost==1.2.2 certifi==2023.11.17 @@ -22,6 +23,7 @@ filelock==3.13.1 fonttools==4.46.0 frozenlist==1.4.0 fsspec==2023.10.0 +gekko==1.0.6 graphviz==0.20.1 hjson==3.1.0 huggingface-hub==0.19.4 @@ -96,6 +98,7 @@ typing-inspect==0.9.0 typing_extensions==4.9.0 tzdata==2023.3 urllib3==2.1.0 +xformers==0.0.22; sys_platform == 'linux' xgboost==2.0.2 xxhash==3.4.1 yarl==1.9.4