Commit 3f7bc418 authored by novelailab's avatar novelailab

Disable deepspeed

parent 4acb13e1
......@@ -31,10 +31,10 @@ RUN pip3 install min-dalle
#basedformer
RUN pip3 install https://www.dropbox.com/s/8ozhhbo1g7y5dsz/basedformer-f7c8c4fe12f8a0acf6588d8d09a8b9b0481895e3.zip?dl=1
#built DS
RUN pip3 install https://www.dropbox.com/s/euzpgpfrs9isf1z/deepspeed-0.7.3%2B55b7b9e0-cp38-cp38-linux_x86_64.whl?dl=1
#RUN pip3 install https://www.dropbox.com/s/euzpgpfrs9isf1z/deepspeed-0.7.3%2B55b7b9e0-cp38-cp38-linux_x86_64.whl?dl=1
RUN pip3 install faiss-cpu sentence_transformers
RUN pip3 install 'git+https://github.com/facebookresearch/xformers@51dd119#egg=xformers'
RUN pip3 uninstall -y triton
#RUN pip install -U --pre triton
#Open ports
EXPOSE 8080
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment