gradio-llama-cpp-2

Sleeping

freddyaboulton HF staff commited on Oct 1

Commit

279f217

•

1 Parent(s): 20f53f9

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -23,8 +23,8 @@ RUN python3 -m pip install --upgrade pip pytest cmake \
     pydantic-settings starlette-context gradio huggingface_hub[hf_transfer] hf_transfer
 # Install llama-cpp-python (build with cuda)
-RUN pip install llama-cpp-python \
-  --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu124
 RUN useradd -m -u 1000 user
 # Switch to the "user" user

     pydantic-settings starlette-context gradio huggingface_hub[hf_transfer] hf_transfer
 # Install llama-cpp-python (build with cuda)
+pip install llama-cpp-python \
+  --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cpu
 RUN useradd -m -u 1000 user
 # Switch to the "user" user