Pratik Dwivedi commited on
Commit
da2bc01
1 Parent(s): 118a2e7

Fix Imports (#5)

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -4,7 +4,7 @@ WORKDIR /code
4
 
5
  COPY . /code
6
 
7
- RUN pip install packaging
8
 
9
  RUN pip install --no-cache-dir torch==2.2.1 --index-url https://download.pytorch.org/whl/cu121
10
 
 
4
 
5
  COPY . /code
6
 
7
+ RUN pip install packaging ninja buildtools
8
 
9
  RUN pip install --no-cache-dir torch==2.2.1 --index-url https://download.pytorch.org/whl/cu121
10