postgresml
diff --git a/‎pgml-extension/Dockerfile
Copy file name to clipboardExpand all lines: pgml-extension/Dockerfile
+2Lines changed: 2 additions & 0 deletions b/‎pgml-extension/Dockerfile
Copy file name to clipboardExpand all lines: pgml-extension/Dockerfile
+2Lines changed: 2 additions & 0 deletions
diff --git a/‎pgml-extension/requirements-xformers.txt
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions b/‎pgml-extension/requirements-xformers.txt
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions
diff --git a/‎pgml-extension/requirements.txt
Copy file name to clipboardExpand all lines: pgml-extension/requirements.txt
-1Lines changed: 0 additions & 1 deletion b/‎pgml-extension/requirements.txt
Copy file name to clipboardExpand all lines: pgml-extension/requirements.txt
-1Lines changed: 0 additions & 1 deletion
@@ -24,6 +24,8 @@ RUN apt-get update && apt-get install -y postgresql-pgml-14
 COPY --chown=postgres:postgres . /app
 WORKDIR /app
 RUN pip3 install -r requirements.txt
+RUN pip3 install -r requirements-xformers.txt --no-dependencies
+
 # Listen on 0.0.0.0 and allow 'root' to connect without a password.
 # Please modify for production deployments accordingly.
 RUN cp /app/docker/postgresql.conf /etc/postgresql/14/main/postgresql.conf
 
@@ -0,0 +1 @@
+xformers==0.0.20
@@ -18,7 +18,6 @@ torchaudio==2.0.2
 torchvision==0.15.2
 tqdm==4.65.0
 transformers==4.29.2
-xformers==0.0.20
 xgboost==1.7.5
 langchain==0.0.180
 einops==0.6.1