gradio-llama-cpp-2

Sleeping

pngwn HF staff commited on May 11

Commit

0254667

•

1 Parent(s): 42b96f4

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -20,7 +20,7 @@ ENV LLAMA_CUBLAS=1
 # Install depencencies
 RUN python3 -m pip install --upgrade pip pytest cmake \
     scikit-build setuptools fastapi uvicorn sse-starlette \
-    pydantic-settings starlette-context gradio huggingface_hub hf_transfer
 # Install llama-cpp-python (build with cuda)
 RUN pip install llama-cpp-python \
@@ -39,6 +39,7 @@ ENV HOME=/home/user \
 	GRADIO_SERVER_NAME=0.0.0.0 \
 	GRADIO_THEME=huggingface \
 	SYSTEM=spaces
 WORKDIR $HOME/app

 # Install depencencies
 RUN python3 -m pip install --upgrade pip pytest cmake \
     scikit-build setuptools fastapi uvicorn sse-starlette \
+    pydantic-settings starlette-context gradio huggingface_hub[hf_transfer] hf_transfer
 # Install llama-cpp-python (build with cuda)
 RUN pip install llama-cpp-python \
 	GRADIO_SERVER_NAME=0.0.0.0 \
 	GRADIO_THEME=huggingface \
 	SYSTEM=spaces
+    HF_HUB_ENABLE_HF_TRANSFER=1
 WORKDIR $HOME/app