Update Dockerfile

Dockerfile

@@ -26,7 +26,9 @@ RUN python3 -m pip install --upgrade pip pytest cmake \
 
 # Install llama-cpp-python (build with cuda)
 # RUN CMAKE_ARGS="-DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=75" FORCE_CMAKE=1 python3 -m pip install llama-cpp-python --force-reinstall --upgrade --no-cache-dir --verbose
-RUN python3 -m pip install llama-cpp-python
+# RUN python3 -m pip install llama-cpp-python
+RUN python3 -m pip install llama-cpp-python \
+  --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
 
 RUN useradd -m -u 1000 user
 # Switch to the "user" user