matthoffner commited on
Commit
b8e402b
·
1 Parent(s): b6217d2

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -4,7 +4,8 @@ ENV DEBIAN_FRONTEND=noninteractive
4
 
5
  # Install dependencies
6
  RUN apt update && \
7
- apt install --no-install-recommends -y build-essential python3 python3-pip wget curl git cmake zlib1g-dev libblas-dev libcublas-dev && \
 
8
  apt clean && rm -rf /var/lib/apt/lists/*
9
 
10
  WORKDIR /app
@@ -37,4 +38,4 @@ EXPOSE 8080
37
  RUN ls -al
38
 
39
  # Start the llava-server with models
40
- CMD ["/app/llava-cpp-server/bin/llava-server", "-m", "/home/user/app/ggml-model-q4_k.gguf", "--mmproj", "/home/user/app/mmproj-model-f16.gguf", "--host", "0.0.0.0", "--port", "8080"]
 
4
 
5
  # Install dependencies
6
  RUN apt update && \
7
+ apt install --no-install-recommends -y build-essential python3 python3-pip wget curl git cmake zlib1g-dev libblas-dev && \
8
+ apt install -y cuda-libraries-dev-12-2 && \
9
  apt clean && rm -rf /var/lib/apt/lists/*
10
 
11
  WORKDIR /app
 
38
  RUN ls -al
39
 
40
  # Start the llava-server with models
41
+ CMD ["/app/llava-cpp-server/bin/llava-server", "-m", "/home/user/app/ggml-model-q4_k.gguf", "--mmproj", "/home/user/app/mmproj-model-f16.gguf", "--host", "0.0.0