Commit 76356352 authored by nanahira's avatar nanahira

tune dockerfile

parent fe89462a
...@@ -8,49 +8,43 @@ RUN apt-get update && apt-get install -y \ ...@@ -8,49 +8,43 @@ RUN apt-get update && apt-get install -y \
python3-virtualenv python3-distutils python3-numpy tini && \ python3-virtualenv python3-distutils python3-numpy tini && \
update-alternatives --install /usr/bin/python python /usr/bin/python3 1 && \ update-alternatives --install /usr/bin/python python /usr/bin/python3 1 && \
update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1 && \ update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1 && \
pip3 install --no-cache-dir --upgrade pip pip3 install --no-cache-dir --upgrade pip && \
#Install Torch(chonky!) #Install Torch(chonky!)
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 && \
pip3 install --no-cache-dir dotmap icecream sentry-sdk numpy fastapi "uvicorn[standard]" gunicorn omegaconf transformers einops && \
#Install Python deps
RUN pip3 install --no-cache-dir dotmap icecream sentry-sdk numpy fastapi "uvicorn[standard]" gunicorn
RUN pip3 install omegaconf transformers einops
#RUN git clone https://github.com/NovelAI/stable-diffusion #RUN git clone https://github.com/NovelAI/stable-diffusion
#RUN pip3 install -e stable-diffusion/. #RUN pip3 install -e stable-diffusion/.
RUN pip3 install pytorch_lightning pip3 install pytorch_lightning && \
RUN pip3 install -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers pip3 install -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
#RUN wget -O stable-diffusion-private-hypernets.zip https://www.dropbox.com/s/lbxm6oijjawmeew/stable-diffusion-private-buckets.zip?dl=1 #RUN wget -O stable-diffusion-private-hypernets.zip https://www.dropbox.com/s/lbxm6oijjawmeew/stable-diffusion-private-buckets.zip?dl=1
#RUN unzip stable-diffusion-private-hypernets.zip #RUN unzip stable-diffusion-private-hypernets.zip
ADD stable-diffusion-private stable-diffusion-private ADD stable-diffusion-private stable-diffusion-private
RUN pip3 install -e stable-diffusion-private/ RUN pip3 install -e stable-diffusion-private/ && \
RUN pip3 install https://github.com/NovelAI/k-diffusion-multigen/archive/457560c53c344b16cf3dda5eb0ba44d83e0b7c46.zip pip3 install https://github.com/NovelAI/k-diffusion-multigen/archive/457560c53c344b16cf3dda5eb0ba44d83e0b7c46.zip && \
RUN pip3 install simplejpeg pip3 install simplejpeg min-dalle && \
RUN pip3 install min-dalle
#RUN pip3 install https://github.com/microsoft/DeepSpeed/archive/55b7b9e008943b8b93d4903d90b255313bb9d82c.zip #RUN pip3 install https://github.com/microsoft/DeepSpeed/archive/55b7b9e008943b8b93d4903d90b255313bb9d82c.zip
#basedformer #basedformer
RUN pip3 install https://www.dropbox.com/s/8ozhhbo1g7y5dsz/basedformer-f7c8c4fe12f8a0acf6588d8d09a8b9b0481895e3.zip?dl=1 pip3 install https://www.dropbox.com/s/8ozhhbo1g7y5dsz/basedformer-f7c8c4fe12f8a0acf6588d8d09a8b9b0481895e3.zip?dl=1 && \
#built DS #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 pip3 install faiss-cpu sentence_transformers && \
RUN wget https://f004.backblazeb2.com/file/naipublicbucketxyz/xformers.zip wget https://f004.backblazeb2.com/file/naipublicbucketxyz/xformers.zip && unzip xformers.zip && \
RUN unzip xformers.zip cd xformers && \
WORKDIR xformers python3 setup.py develop && \
RUN python3 setup.py develop pip uninstall -y pillow && \
RUN pip uninstall -y pillow pip install pillow-simd
RUN pip install pillow-simd
#Open ports #Open ports
EXPOSE 8080 #EXPOSE 8080
EXPOSE 80 #EXPOSE 80
EXPOSE 443 #EXPOSE 443
EXPOSE 4369 #EXPOSE 4369
EXPOSE 5672 #EXPOSE 5672
EXPOSE 25672 #EXPOSE 25672
EXPOSE 15672 #EXPOSE 15672
EXPOSE 15692 #EXPOSE 15692
EXPOSE 50051 #EXPOSE 50051
#Copy node src and run #Copy node src and run
WORKDIR /usr/src/app WORKDIR /usr/src/app
COPY . . COPY . .
......
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