diff --git a/Docker/cuda/cuda.Dockerfile b/Docker/cuda/cuda.Dockerfile index 36bff3a8..2b76df12 100644 --- a/Docker/cuda/cuda.Dockerfile +++ b/Docker/cuda/cuda.Dockerfile @@ -47,4 +47,4 @@ RUN CMAKE_ARGS="-DLLAMA_CUDA=on" FORCE_CMAKE=1 pip install 'llama-cpp-python[ser EXPOSE 8008 # Run the server -CMD python3 -m llama_cpp.server --config_file config-cuda.json +CMD ["python", "-m", "llama_cpp.server", "--config_file", "config-cuda.json"]