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