infinity-emb-mxbai

Runtime error

jerukperas commited on Sep 14

Commit

1399419

•

1 Parent(s): 6fa344f

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -9,5 +9,5 @@ ENV PATH="/home/user/.local/bin:$PATH"
 WORKDIR /infinity_emb
 RUN pip install --no-cache-dir -qU pip
-RUN pip install --no-cache-dir -qU 'infinity-emb[server,cache,logging,einops,torch]'
 CMD ["infinity_emb", "v2", "--host", "0.0.0.0", "--port", "7860"]

 WORKDIR /infinity_emb
 RUN pip install --no-cache-dir -qU pip
+RUN pip install --no-cache-dir -qU 'infinity-emb[server,cache,logging,einops,torch,optimum]'
 CMD ["infinity_emb", "v2", "--host", "0.0.0.0", "--port", "7860"]