novaco-evildudes999 commited on
Commit
7e00a21
·
verified ·
1 Parent(s): 70e1801

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +11 -3
Dockerfile CHANGED
@@ -1,15 +1,24 @@
1
  FROM python:3.10-slim
2
 
3
  WORKDIR /app
 
4
  ENV DEBIAN_FRONTEND=noninteractive
5
  ENV PYTHONUNBUFFERED=1
6
  ENV HF_HOME=/tmp/hf_cache
7
- ENV PIP_NO_CACHE_DIR=1
 
 
8
 
9
  RUN apt-get update && apt-get install -y --no-install-recommends \
 
 
 
 
10
  ca-certificates \
11
  && rm -rf /var/lib/apt/lists/*
12
 
 
 
13
  RUN pip install --upgrade pip setuptools wheel
14
 
15
  RUN pip install --no-cache-dir \
@@ -17,8 +26,7 @@ RUN pip install --no-cache-dir \
17
  "uvicorn[standard]" \
18
  huggingface-hub \
19
  pydantic \
20
- llama-cpp-python \
21
- --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cpu
22
 
23
  COPY . .
24
 
 
1
  FROM python:3.10-slim
2
 
3
  WORKDIR /app
4
+
5
  ENV DEBIAN_FRONTEND=noninteractive
6
  ENV PYTHONUNBUFFERED=1
7
  ENV HF_HOME=/tmp/hf_cache
8
+ ENV CC=/usr/bin/gcc
9
+ ENV CXX=/usr/bin/g++
10
+ ENV CMAKE_ARGS="-DGGML_BLAS=ON -DGGML_BLAS_VENDOR=OpenBLAS"
11
 
12
  RUN apt-get update && apt-get install -y --no-install-recommends \
13
+ build-essential \
14
+ cmake \
15
+ pkg-config \
16
+ libopenblas-dev \
17
  ca-certificates \
18
  && rm -rf /var/lib/apt/lists/*
19
 
20
+ RUN gcc --version && g++ --version
21
+
22
  RUN pip install --upgrade pip setuptools wheel
23
 
24
  RUN pip install --no-cache-dir \
 
26
  "uvicorn[standard]" \
27
  huggingface-hub \
28
  pydantic \
29
+ llama-cpp-python
 
30
 
31
  COPY . .
32