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

Fix Packaging (#4)

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -0
  2. requirements.txt +0 -1
Dockerfile CHANGED
@@ -4,6 +4,8 @@ WORKDIR /code
4
 
5
  COPY . /code
6
 
 
 
7
  RUN pip install --no-cache-dir torch==2.2.1 --index-url https://download.pytorch.org/whl/cu121
8
 
9
  RUN pip install --no-cache-dir -r /code/requirements.txt
 
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
 
11
  RUN pip install --no-cache-dir -r /code/requirements.txt
requirements.txt CHANGED
@@ -1,4 +1,3 @@
1
- packaging
2
  transformers
3
  causal-conv1d
4
  mamba-ssm
 
 
1
  transformers
2
  causal-conv1d
3
  mamba-ssm