pin llama cpp python

Smartappli · Jul 27, 2024 · cb09745 · cb09745
1 parent f8adc19
commit cb09745
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Docker/cuda/cuda.Dockerfile b/Docker/cuda/cuda.Dockerfile
@@ -27,9 +27,9 @@ RUN python3 -m venv venv
 # RUN python3 -m pip install --upgrade pip pytest cmake scikit-build setuptools fastapi uvicorn sse-starlette pydantic-settings starlette-context
 
 # Install llama-cpp-python (build with cuda)
-RUN CMAKE_ARGS="-DGGML_CUDA=on" venv/bin/pip install .[server]
-# RUN make clean
+RUN CMAKE_ARGS="-DGGML_CUDA=on" venv/bin/pip install .[server]==0.2.83
 
+# RUN make clean
 FROM nvidia/cuda:${CUDA_RUNTIME_IMAGE} as runtime
 
 # We need to set the host to 0.0.0.0 to allow outside access