Update Dockerfile

Dockerfile

@@ -7,7 +7,10 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 # RUN CMAKE_ARGS="-DLLAMA_CUBLAS=off" pip install llama-cpp-python
 
 RUN pip install --upgrade --quiet  llama-cpp-python
-RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" FORCE_CMAKE=1 pip install --upgrade --force-reinstall llama-cpp-python --no-cache-dir
+ENV CMAKE_ARGS="-DLLAMA_CUBLAS=on"
+RUN FORCE_CMAKE=1 pip install --upgrade --force-reinstall llama-cpp-python --no-cache-dir
+
+
 #RUN CMAKE_ARGS="-DLLAMA_BLAS=ON -DLLAMA_BLAS_VENDOR=OpenBLAS" pip install llama-cpp-python