From 2d8c034df3d0b47b89c58ba12a1e9e6a00c9b089 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Mon, 25 Sep 2023 09:58:52 +0000
Subject: [PATCH] Adding target list.

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 6672a85a..f29fa107 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -121,7 +121,7 @@ FROM kernel-builder as awq-kernels-builder
 WORKDIR /usr/src
 COPY server/Makefile-awq Makefile
 # Build specific version of transformers
-RUN make build-awq
+RUN TORCH_CUDA_ARCH_LIST="8.0;8.6" make build-awq
 
 # Build Transformers CUDA kernels
 FROM kernel-builder as custom-kernels-builder