From 2d6c140c478b86a4cb46b6bda07d84112e2759ee Mon Sep 17 00:00:00 2001 From: acmo0 Date: Tue, 28 May 2024 15:42:25 +0200 Subject: [PATCH] Add model download in dockerfile --- Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/Dockerfile b/Dockerfile index 0d5e697..9af5e47 100644 --- a/Dockerfile +++ b/Dockerfile @@ -10,6 +10,8 @@ RUN pip install --no-cache-dir -r requirements_api.txt COPY . . +wget -O app/models/models_weight/ld-model.ckpt https://ommer-lab.com/files/latent-diffusion/nitro/txt2img-f8-large/model.ckpt + EXPOSE 5000 CMD ["uvicorn", "app.main:app", "--host","0.0.0.0", "--port", "5000"] \ No newline at end of file