acharyaaditya26 commited on
Commit
6b5f0df
1 Parent(s): 586c40f

docker changes

Browse files
Files changed (1) hide show
  1. Dockerfile +9 -0
Dockerfile CHANGED
@@ -7,6 +7,9 @@ RUN apt-get update && \
7
  apt-get clean && \
8
  rm -rf /var/lib/apt/lists/*
9
 
 
 
 
10
  # Create a user with UID 1000
11
  RUN useradd -m -u 1000 user
12
  USER user
@@ -20,6 +23,12 @@ COPY --chown=user ./requirements.txt requirements.txt
20
  RUN pip3 install --upgrade pip
21
  RUN pip3 install packaging
22
  RUN pip3 install --no-cache-dir --upgrade -r requirements.txt
 
 
 
 
 
 
23
  RUN pip3 install flash_attn
24
 
25
  # Copy the rest of the application
 
7
  apt-get clean && \
8
  rm -rf /var/lib/apt/lists/*
9
 
10
+ # Install git
11
+ RUN apt-get update && apt-get install -y git
12
+
13
  # Create a user with UID 1000
14
  RUN useradd -m -u 1000 user
15
  USER user
 
23
  RUN pip3 install --upgrade pip
24
  RUN pip3 install packaging
25
  RUN pip3 install --no-cache-dir --upgrade -r requirements.txt
26
+
27
+
28
+ # Install numpy version compatible with flash_attn
29
+ RUN pip3 install numpy==1.23.4
30
+
31
+ # Install flash_attn
32
  RUN pip3 install flash_attn
33
 
34
  # Copy the rest of the application