devve1 commited on
Commit
27343fd
1 Parent(s): d7c3080

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -5,7 +5,7 @@ ARG DEBIAN_FRONTEND=noninteractive
5
  RUN apt-get update && apt-get install --no-install-recommends -y \
6
  curl \
7
  build-essential \
8
- python3.9 \
9
  python3-pip \
10
  git \
11
  && apt-get clean && rm -rf /var/lib/apt/lists/*
@@ -15,6 +15,7 @@ WORKDIR /app
15
  COPY ./requirements.txt /app/requirements.txt
16
 
17
  RUN pip3 install --no-cache-dir --upgrade -r /app/requirements.txt
 
18
 
19
  RUN useradd -m -u 1000 user
20
  USER user
 
5
  RUN apt-get update && apt-get install --no-install-recommends -y \
6
  curl \
7
  build-essential \
8
+ python3.10 \
9
  python3-pip \
10
  git \
11
  && apt-get clean && rm -rf /var/lib/apt/lists/*
 
15
  COPY ./requirements.txt /app/requirements.txt
16
 
17
  RUN pip3 install --no-cache-dir --upgrade -r /app/requirements.txt
18
+ RUN pip3 install flash-attn --no-build-isolation --no-cache-dir
19
 
20
  RUN useradd -m -u 1000 user
21
  USER user