Update cuda.Dockerfile

Smartappli · Jul 26, 2024 · 43962f0 · 43962f0
1 parent 5eb46aa
commit 43962f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"]