From b9775c489344d7d0da35a0a343552dbd7896146e Mon Sep 17 00:00:00 2001 From: vuonghoainam Date: Wed, 30 Aug 2023 20:07:42 +0700 Subject: [PATCH] fix(docker): Update dockerfile for sd ggml api --- docker-compose.yml | 11 ++++++----- jan-inference/sd/inference.Dockerfile | 4 +++- 2 files changed, 9 insertions(+), 6 deletions(-) diff --git a/docker-compose.yml b/docker-compose.yml index 5b456f33e..757691320 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -190,15 +190,16 @@ services: dockerfile: inference.Dockerfile # Mount the directory that contains the downloaded model. volumes: - - ./jan-inference/sd/models:/models - - ./jan-inference/sd/output/:/serving/output + - ./jan-inference/sd/models:/models/ + - ./jan-inference/sd/output/:/output/ command: /bin/bash -c "python -m uvicorn main:app --proxy-headers --host 0.0.0.0 --port 8000" environment: # Specify the path to the model for the web application. - BASE_URL: http://0.0.0.0:8000 + BASE_URL: http://0.0.0.0:8001 MODEL_NAME: ${SD_MODEL_FILE}.q4_0.bin - MODEL_DIR: "/models" - SD_PATH: "/sd" + MODEL_DIR: /models + OUTPUT_DIR: /output + SD_PATH: /sd PYTHONUNBUFFERED: 1 ports: - 8001:8000 diff --git a/jan-inference/sd/inference.Dockerfile b/jan-inference/sd/inference.Dockerfile index 39d42dd43..db31e7f98 100644 --- a/jan-inference/sd/inference.Dockerfile +++ b/jan-inference/sd/inference.Dockerfile @@ -1,4 +1,6 @@ -FROM python:3.9.17 as build +ARG UBUNTU_VERSION=22.04 + +FROM ubuntu:$UBUNTU_VERSION as build RUN apt-get update && apt-get install -y build-essential git cmake