From 9740a9be435314fe04b1de109bef00bb5272319e Mon Sep 17 00:00:00 2001 From: JimothyJohn Date: Sat, 23 Nov 2024 13:01:13 -0600 Subject: [PATCH] Fix Dockerfile --- Dockerfile | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index f73beff..dc317e7 100755 --- a/Dockerfile +++ b/Dockerfile @@ -1,20 +1,28 @@ FROM nvcr.io/nvidia/pytorch:24.06-py3 +ENV PATH=/opt/conda/bin:$PATH + +RUN apt-get update && apt-get install -y \ + libgl1-mesa-glx \ + libglib2.0-0 \ + && rm -rf /var/lib/apt/lists/* + WORKDIR /app RUN curl https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -o ~/miniconda.sh \ && sh ~/miniconda.sh -b -p /opt/conda \ && rm ~/miniconda.sh -ENV PATH /opt/conda/bin:$PATH COPY pyproject.toml pyproject.toml COPY diffusion diffusion COPY configs configs COPY sana sana COPY app app +COPY tools tools COPY environment_setup.sh environment_setup.sh RUN ./environment_setup.sh sana # COPY server.py server.py -CMD ["conda", "run", "-n", "sana", "--no-capture-output", "python", "-u", "-W", "ignore", "app/app_sana.py", "--config=configs/sana_config/1024ms/Sana_1600M_img1024.yaml", "--model_path=hf://Sana_1600M_1024px/checkpoints/Sana_1600M_1024px.pth",] +CMD ["conda", "run", "-n", "sana", "--no-capture-output", "python", "-u", "-W", "ignore", "app/app_sana.py", "--config=configs/sana_config/1024ms/Sana_1600M_img1024.yaml", "--model_path=hf://Efficient-Large-Model/Sana_1600M_1024px/checkpoints/Sana_1600M_1024px.pth"] +