forked from zylon-ai/private-gpt
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
# Conflicts: # .github/workflows/generate-release.yml
- Loading branch information
Showing
13 changed files
with
321 additions
and
42 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,16 @@ | ||
| http: | ||
| services: | ||
| ollama: | ||
| loadBalancer: | ||
| healthCheck: | ||
| interval: 5s | ||
| path: / | ||
| servers: | ||
| - url: http://ollama-cpu:11434 | ||
| - url: http://ollama-cuda:11434 | ||
| - url: http://host.docker.internal:11434 | ||
|
|
||
| routers: | ||
| ollama-router: | ||
| rule: "PathPrefix(`/`)" | ||
| service: ollama |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
File renamed without changes.
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,19 +1,102 @@ | ||
| services: | ||
| private-gpt: | ||
|
|
||
| #----------------------------------- | ||
| #---- Private-GPT services --------- | ||
| #----------------------------------- | ||
|
|
||
| # Private-GPT service for the Ollama CPU and GPU modes | ||
| # This service builds from an external Dockerfile and runs the Ollama mode. | ||
| private-gpt-ollama: | ||
| image: ${PGPT_IMAGE:-zylonai/private-gpt}${PGPT_TAG:-0.6.1}-ollama | ||
| build: | ||
| dockerfile: Dockerfile.external | ||
| context: . | ||
| dockerfile: Dockerfile.ollama | ||
| volumes: | ||
| - ./local_data/:/home/worker/app/local_data | ||
| ports: | ||
| - 8001:8001 | ||
| - "8001:8001" | ||
| environment: | ||
| PORT: 8001 | ||
| PGPT_PROFILES: docker | ||
| PGPT_MODE: ollama | ||
| PGPT_EMBED_MODE: ollama | ||
| PGPT_OLLAMA_API_BASE: http://ollama:11434 | ||
| HF_TOKEN: ${HF_TOKEN:-} | ||
| profiles: | ||
| - "" | ||
| - ollama-cpu | ||
| - ollama-cuda | ||
| - ollama-api | ||
|
|
||
| # Private-GPT service for the local mode | ||
| # This service builds from a local Dockerfile and runs the application in local mode. | ||
| private-gpt-llamacpp-cpu: | ||
| image: ${PGPT_IMAGE:-zylonai/private-gpt}${PGPT_TAG:-0.6.1}-llamacpp-cpu | ||
| build: | ||
| context: . | ||
| dockerfile: Dockerfile.llamacpp-cpu | ||
| volumes: | ||
| - ./local_data/:/home/worker/app/local_data | ||
| - ./models/:/home/worker/app/models | ||
| entrypoint: sh -c ".venv/bin/python scripts/setup && .venv/bin/python -m private_gpt" | ||
| ports: | ||
| - "8001:8001" | ||
| environment: | ||
| PORT: 8001 | ||
| PGPT_PROFILES: local | ||
| HF_TOKEN: ${HF_TOKEN} | ||
| profiles: | ||
| - llamacpp-cpu | ||
|
|
||
| #----------------------------------- | ||
| #---- Ollama services -------------- | ||
| #----------------------------------- | ||
|
|
||
| # Traefik reverse proxy for the Ollama service | ||
| # This will route requests to the Ollama service based on the profile. | ||
| ollama: | ||
| image: ollama/ollama:latest | ||
| image: traefik:v2.10 | ||
| ports: | ||
| - 11434:11434 | ||
| - "11435:11434" | ||
| - "8081:8080" | ||
| command: | ||
| - "--providers.file.filename=/etc/router.yml" | ||
| - "--log.level=ERROR" | ||
| - "--api.insecure=true" | ||
| - "--providers.docker=true" | ||
| - "--providers.docker.exposedbydefault=false" | ||
| - "--entrypoints.web.address=:11434" | ||
| volumes: | ||
| - /var/run/docker.sock:/var/run/docker.sock:ro | ||
| - ./.docker/router.yml:/etc/router.yml:ro | ||
| extra_hosts: | ||
| - "host.docker.internal:host-gateway" | ||
| profiles: | ||
| - "" | ||
| - ollama-cpu | ||
| - ollama-cuda | ||
| - ollama-api | ||
|
|
||
| # Ollama service for the CPU mode | ||
| ollama-cpu: | ||
| image: ollama/ollama:latest | ||
| volumes: | ||
| - ./models:/root/.ollama | ||
| profiles: | ||
| - "" | ||
| - ollama | ||
|
|
||
| # Ollama service for the CUDA mode | ||
| ollama-cuda: | ||
| image: ollama/ollama:latest | ||
| volumes: | ||
| - ./models:/root/.ollama | ||
| deploy: | ||
| resources: | ||
| reservations: | ||
| devices: | ||
| - driver: nvidia | ||
| count: 1 | ||
| capabilities: [gpu] | ||
| profiles: | ||
| - ollama-cuda |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,105 @@ | ||
| This guide provides a quick start for running different profiles of PrivateGPT using Docker Compose. | ||
| The profiles cater to various environments, including Ollama setups (CPU, CUDA, MacOS), and a fully local setup. | ||
|
|
||
| By default, Docker Compose will download pre-built images from a remote registry when starting the services. However, you have the option to build the images locally if needed. Details on building Docker image locally are provided at the end of this guide. | ||
|
|
||
| If you want to run PrivateGPT locally without Docker, refer to the [Local Installation Guide](/installation). | ||
|
|
||
| ## Prerequisites | ||
| - **Docker and Docker Compose:** Ensure both are installed on your system. | ||
| [Installation Guide for Docker](https://docs.docker.com/get-docker/), [Installation Guide for Docker Compose](https://docs.docker.com/compose/install/). | ||
| - **Clone PrivateGPT Repository:** Clone the PrivateGPT repository to your machine and navigate to the directory: | ||
| ```sh | ||
| git clone https://github.com/zylon-ai/private-gpt.git | ||
| cd private-gpt | ||
| ``` | ||
|
|
||
| ## Setups | ||
|
|
||
| ### Ollama Setups (Recommended) | ||
|
|
||
| #### 1. Default/Ollama CPU | ||
|
|
||
| **Description:** | ||
| This profile runs the Ollama service using CPU resources. It is the standard configuration for running Ollama-based Private-GPT services without GPU acceleration. | ||
|
|
||
| **Run:** | ||
| To start the services using pre-built images, run: | ||
| ```sh | ||
| docker-compose up | ||
| ``` | ||
| or with a specific profile: | ||
| ```sh | ||
| docker-compose --profile ollama-cpu up | ||
| ``` | ||
|
|
||
| #### 2. Ollama Nvidia CUDA | ||
|
|
||
| **Description:** | ||
| This profile leverages GPU acceleration with CUDA support, suitable for computationally intensive tasks that benefit from GPU resources. | ||
|
|
||
| **Requirements:** | ||
| Ensure that your system has compatible GPU hardware and the necessary NVIDIA drivers installed. The installation process is detailed [here](https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html). | ||
|
|
||
| **Run:** | ||
| To start the services with CUDA support using pre-built images, run: | ||
| ```sh | ||
| docker-compose --profile ollama-cuda up | ||
| ``` | ||
|
|
||
| #### 3. Ollama External API | ||
|
|
||
| **Description:** | ||
| This profile is designed for running PrivateGPT using Ollama installed on the host machine. This setup is particularly useful for MacOS users, as Docker does not yet support Metal GPU. | ||
|
|
||
| **Requirements:** | ||
| Install Ollama on your machine by following the instructions at [ollama.ai](https://ollama.ai/). | ||
|
|
||
| **Run:** | ||
| To start the Ollama service, use: | ||
| ```sh | ||
| OLLAMA_HOST=0.0.0.0 ollama serve | ||
| ``` | ||
| To start the services with the host configuration using pre-built images, run: | ||
| ```sh | ||
| docker-compose --profile ollama-api up | ||
| ``` | ||
|
|
||
| ### Fully Local Setups | ||
|
|
||
| #### 1. LlamaCPP CPU | ||
|
|
||
| **Description:** | ||
| This profile runs the Private-GPT services locally using `llama-cpp` and Hugging Face models. | ||
|
|
||
| **Requirements:** | ||
| A **Hugging Face Token (HF_TOKEN)** is required for accessing Hugging Face models. Obtain your token following [this guide](/installation/getting-started/troubleshooting#downloading-gated-and-private-models). | ||
|
|
||
| **Run:** | ||
| Start the services with your Hugging Face token using pre-built images: | ||
| ```sh | ||
| HF_TOKEN=<your_hf_token> docker-compose --profile llamacpp-cpu up | ||
| ``` | ||
| Replace `<your_hf_token>` with your actual Hugging Face token. | ||
|
|
||
| ## Building Locally | ||
|
|
||
| If you prefer to build Docker images locally, which is useful when making changes to the codebase or the Dockerfiles, follow these steps: | ||
|
|
||
| ### Building Locally | ||
| To build the Docker images locally, navigate to the cloned repository directory and run: | ||
| ```sh | ||
| docker-compose build | ||
| ``` | ||
| This command compiles the necessary Docker images based on the current codebase and Dockerfile configurations. | ||
|
|
||
| ### Forcing a Rebuild with --build | ||
| If you have made changes and need to ensure these changes are reflected in the Docker images, you can force a rebuild before starting the services: | ||
| ```sh | ||
| docker-compose up --build | ||
| ``` | ||
| or with a specific profile: | ||
| ```sh | ||
| docker-compose --profile <profile_name> up --build | ||
| ``` | ||
| Replace `<profile_name>` with the desired profile. |
Oops, something went wrong.