Skip to content

Commit

Permalink
fix: add built image from DockerHub (#2042)
Browse files Browse the repository at this point in the history
* chore: update docker-compose with profiles

* docs: add quick start doc

* chore: generate docker release when new version is released

* chore: add dockerhub image in docker-compose

* docs: update quickstart with local/remote images

* chore: update docker tag

* chore: refactor dockerfile names

* chore: update docker-compose names

* docs: update llamacpp naming

* fix: naming

* docs: fix llamacpp command
  • Loading branch information
jaluma authored Aug 5, 2024
1 parent 1c665f7 commit f09f6dd
Show file tree
Hide file tree
Showing 5 changed files with 62 additions and 39 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/generate-release.yml
Original file line number Diff line number Diff line change
Expand Up @@ -9,15 +9,15 @@ env:
REGISTRY: docker.io
IMAGE_NAME: ${{ github.repository }}
platforms: linux/amd64,linux/arm64
DEFAULT_TYPE: "external"
DEFAULT_TYPE: "ollama"

jobs:
build-and-push-image:
runs-on: ubuntu-latest

strategy:
matrix:
type: [ local, external ]
type: [ llamacpp-cpu, ollama ]

permissions:
contents: read
Expand Down
File renamed without changes.
File renamed without changes.
19 changes: 11 additions & 8 deletions docker-compose.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,10 @@ 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:
context: .
dockerfile: Dockerfile.external
dockerfile: Dockerfile.ollama
volumes:
- ./local_data/:/home/worker/app/local_data
ports:
Expand All @@ -20,18 +21,20 @@ services:
PGPT_MODE: ollama
PGPT_EMBED_MODE: ollama
PGPT_OLLAMA_API_BASE: http://ollama:11434
HF_TOKEN: ${HF_TOKEN:-}
profiles:
- ""
- ollama
- ollama-cpu
- ollama-cuda
- ollama-host
- 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-local:
private-gpt-llamacpp-cpu:
image: ${PGPT_IMAGE:-zylonai/private-gpt}${PGPT_TAG:-0.6.1}-llamacpp-cpu
build:
context: .
dockerfile: Dockerfile.local
dockerfile: Dockerfile.llamacpp-cpu
volumes:
- ./local_data/:/home/worker/app/local_data
- ./models/:/home/worker/app/models
Expand All @@ -43,7 +46,7 @@ services:
PGPT_PROFILES: local
HF_TOKEN: ${HF_TOKEN}
profiles:
- local
- llamacpp-cpu

#-----------------------------------
#---- Ollama services --------------
Expand All @@ -70,9 +73,9 @@ services:
- "host.docker.internal:host-gateway"
profiles:
- ""
- ollama
- ollama-cpu
- ollama-cuda
- ollama-host
- ollama-api

# Ollama service for the CPU mode
ollama-cpu:
Expand Down
78 changes: 49 additions & 29 deletions fern/docs/pages/quickstart/quickstart.mdx
Original file line number Diff line number Diff line change
@@ -1,85 +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 fully Local setup.
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
## 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/).
[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
```

---

## Ollama Setups (Recommended)
## Setups

Ollama setups are recommended for their ease of use and optimized configurations. Ollama offers different profiles depending on your hardware capabilities and operating system.
### Ollama Setups (Recommended)

### 1. Default/Ollama CPU
#### 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, use either of the following commands:
To start the services using pre-built images, run:
```sh
docker-compose up
```
or
or with a specific profile:
```sh
docker-compose --profile ollama up
docker-compose --profile ollama-cpu up
```

### 2. Ollama Nvidia CUDA
#### 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).
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, use:
To start the services with CUDA support using pre-built images, run:
```sh
docker-compose --profile ollama-cuda up
```

### 3. Ollama Host
#### 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/).
- Start the Ollama service with the command:
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
```

**Run:**
To start the services with the host configuration, use:
To start the services with the host configuration using pre-built images, run:
```sh
docker-compose --profile ollama-host up
docker-compose --profile ollama-api up
```

---

## Fully Local Setups
### Fully Local Setups

### LlamaCPP + HuggingFace Embeddings
#### 1. LlamaCPP CPU

**Description:**
This profile runs the Private-GPT services locally using `llama-cpp` and Hugging Face models.

**Requirements:**
- **Hugging Face Token (HF_TOKEN):** Required for accessing Hugging Face models. Obtain your token following [this guide](/installation/getting-started/troubleshooting#downloading-gated-and-private-models).
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:
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
HF_TOKEN=<your_hf_token> docker-compose up --profile local
docker-compose --profile <profile_name> up --build
```
Replace `<your_hf_token>` with your actual Hugging Face token.
Replace `<profile_name>` with the desired profile.

0 comments on commit f09f6dd

Please sign in to comment.