From 9bea40bd40f01ecf29e6d13e14507764a3996aa5 Mon Sep 17 00:00:00 2001
From: Joseph Young <130690493+yousecjoe@users.noreply.github.com>
Date: Wed, 20 Mar 2024 18:16:11 -0400
Subject: [PATCH] Adding the missing env variable ENV
 RAG_EMBEDDING_MODEL_DEVICE_TYPE="cuda"

---
 Dockerfile-cuda | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Dockerfile-cuda b/Dockerfile-cuda
index ef4ae3ecc..b8064e5c5 100644
--- a/Dockerfile-cuda
+++ b/Dockerfile-cuda
@@ -32,6 +32,7 @@ RUN npm run build
 ######## CUDA WebUI backend ########
 FROM --platform=linux/amd64 nvidia/cuda:"$CUDA_VERSION"-devel-ubuntu22.04 AS cuda-build-amd64
 #FROM --platform=linux/amd64 cgr.dev/chainguard/pytorch-cuda12:latest AS cuda-build-amd64 # fails with python requirements conflicts
+ENV RAG_EMBEDDING_MODEL_DEVICE_TYPE="cuda"
 
 # Set environment variables for NVIDIA Container Toolkit
 ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 \