From 8625188549306454a0e408570584463fd19b1a1c Mon Sep 17 00:00:00 2001
From: Alexandru Gherghescu <gherghescu_alex1@yahoo.ro>
Date: Wed, 29 Nov 2023 22:30:35 +0200
Subject: [PATCH] Fix ARG to ENV in Dockerfile

---
 Dockerfile-13b-chat | 2 +-
 Dockerfile-70b-chat | 2 +-
 Dockerfile-7b-chat  | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/Dockerfile-13b-chat b/Dockerfile-13b-chat
index bea11af..82ca45e 100644
--- a/Dockerfile-13b-chat
+++ b/Dockerfile-13b-chat
@@ -27,7 +27,7 @@ RUN git clone https://gitlab.cs.pub.ro/netsys/llama /llama
 ADD llama-2-13b-chat/ /llama/llama-2-13b-chat/
 
 # avoid an annoying PyTorch (torch.distributed) warning
-ARG OMP_NUM_THREADS=1
+ENV OMP_NUM_THREADS=1
 
 # run llama example program
 CMD ["torchrun", \
diff --git a/Dockerfile-70b-chat b/Dockerfile-70b-chat
index 9d6f6a2..d485571 100644
--- a/Dockerfile-70b-chat
+++ b/Dockerfile-70b-chat
@@ -27,7 +27,7 @@ RUN git clone https://gitlab.cs.pub.ro/netsys/llama /llama
 ADD llama-2-70b-chat/ /llama/llama-2-70b-chat/
 
 # avoid an annoying PyTorch (torch.distributed) warning
-ARG OMP_NUM_THREADS=1
+ENV OMP_NUM_THREADS=1
 
 # run llama example program
 CMD ["torchrun", \
diff --git a/Dockerfile-7b-chat b/Dockerfile-7b-chat
index 92fe7db..1885ad2 100644
--- a/Dockerfile-7b-chat
+++ b/Dockerfile-7b-chat
@@ -27,7 +27,7 @@ RUN git clone https://gitlab.cs.pub.ro/netsys/llama /llama
 ADD llama-2-7b-chat/ /llama/llama-2-7b-chat/
 
 # avoid an annoying PyTorch (torch.distributed) warning
-ARG OMP_NUM_THREADS=1
+ENV OMP_NUM_THREADS=1
 
 # run llama example program
 CMD ["torchrun", \
-- 
GitLab