diff --git a/Dockerfile b/Dockerfile index 25c9db1..e08dbb0 100644 --- a/Dockerfile +++ b/Dockerfile @@ -5,7 +5,7 @@ ARG CUDA_FROM=nvidia/cuda:$CUDA_VERSION-cudnn$CUDNN_VERSION-devel-ubuntu$UBUNTU_ ARG OLLAMA_VERSION="0.1.22" ARG OLLAMA_FROM=ollama/ollama:$OLLAMA_VERSION -FROM $OLLAMA_FROM as ollama +FROM $OLLAMA_FROM AS ollama # Base NVidia CUDA Ubuntu image FROM $CUDA_FROM