jerukperas commited on
Commit
6fa344f
1 Parent(s): e54c77d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -9,5 +9,5 @@ ENV PATH="/home/user/.local/bin:$PATH"
9
 
10
  WORKDIR /infinity_emb
11
  RUN pip install --no-cache-dir -qU pip
12
- RUN pip install --no-cache-dir -qU 'infinity-emb[server,cache,logging,einops,optimum]'
13
  CMD ["infinity_emb", "v2", "--host", "0.0.0.0", "--port", "7860"]
 
9
 
10
  WORKDIR /infinity_emb
11
  RUN pip install --no-cache-dir -qU pip
12
+ RUN pip install --no-cache-dir -qU 'infinity-emb[server,cache,logging,einops,torch]'
13
  CMD ["infinity_emb", "v2", "--host", "0.0.0.0", "--port", "7860"]