From 4a290742910c79fe3ce110542465946c347a6b0a Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Mon, 25 Sep 2023 09:41:25 +0000
Subject: [PATCH] Update dockerfile with new build.

---
 Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 84b85040..6672a85a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -119,9 +119,9 @@ RUN TORCH_CUDA_ARCH_LIST="8.0;8.6+PTX" python setup.py build
 # Build Transformers awq kernels
 FROM kernel-builder as awq-kernels-builder
 WORKDIR /usr/src
-COPY server/awq_kernels/ .
+COPY server/Makefile-awq Makefile
 # Build specific version of transformers
-RUN python setup.py build
+RUN make build-awq
 
 # Build Transformers CUDA kernels
 FROM kernel-builder as custom-kernels-builder