From 57222497ecd93eac5e0268af6dd4bc4901c9cba7 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Wed, 13 Mar 2024 17:57:45 +0100 Subject: [PATCH] fix(docker-compose): update docker compose file (#1824) Signed-off-by: Ettore Di Giacinto --- .env | 2 +- docker-compose.yaml | 19 +++++++++++++++++-- 2 files changed, 18 insertions(+), 3 deletions(-) diff --git a/.env b/.env index a31e1e37..82a64e3d 100644 --- a/.env +++ b/.env @@ -18,7 +18,7 @@ ## Default path for models # -MODELS_PATH=/models +# MODELS_PATH=/models ## Enable debug mode # DEBUG=true diff --git a/docker-compose.yaml b/docker-compose.yaml index b24d3c04..b6384c52 100644 --- a/docker-compose.yaml +++ b/docker-compose.yaml @@ -2,15 +2,30 @@ version: '3.6' services: api: - image: quay.io/go-skynet/local-ai:latest + # See https://localai.io/basics/getting_started/#container-images for + # a list of available container images (or build your own with the provided Dockerfile) + # Available images with CUDA, ROCm, SYCL + # Image list (quay.io): https://quay.io/repository/go-skynet/local-ai?tab=tags + # Image list (dockerhub): https://hub.docker.com/r/localai/localai + image: quay.io/go-skynet/local-ai:master-ffmpeg-core build: context: . dockerfile: Dockerfile + args: + - IMAGE_TYPE=core + - BASE_IMAGE=ubuntu:22.04 ports: - 8080:8080 env_file: - .env + environment: + - MODELS_PATH=/models + # - DEBUG=true volumes: - ./models:/models:cached - ./images/:/tmp/generated/images/ - command: ["/usr/bin/local-ai" ] + command: + # Here we can specify a list of models to run (see quickstart https://localai.io/basics/getting_started/#running-models ) + # or an URL pointing to a YAML configuration file, for example: + # - https://gist.githubusercontent.com/mudler/ad601a0488b497b69ec549150d9edd18/raw/a8a8869ef1bb7e3830bf5c0bae29a0cce991ff8d/phi-2.yaml + - phi-2