README.md 5.69 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
<!--
SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
SPDX-License-Identifier: Apache-2.0

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
-->

18
# LLM Deployment Examples
19

20
This directory contains examples and reference implementations for deploying Large Language Models (LLMs) in various configurations.
21

22
## Components
23

24
25
26
- workers: Prefill and decode worker handles actual LLM inference
- router: Handles API requests and routes them to appropriate workers based on specified strategy
- frontend: OpenAI compatible http server handles incoming requests
27

28
## Deployment Architectures
29

30
### Aggregated
31
32
33
34
35
Single-instance deployment where both prefill and decode are done by the same worker.

### Disaggregated
Distributed deployment where prefill and decode are done by separate workers that can scale independently.

36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
```mermaid
sequenceDiagram
    participant D as VllmWorker
    participant Q as PrefillQueue
    participant P as PrefillWorker

    Note over D: Request is routed to decode
    D->>D: Decide if prefill should be done locally or remotely

        D->>D: Allocate KV blocks
        D->>Q: Put RemotePrefillRequest on the queue

        P->>Q: Pull request from the queue
        P-->>D: Read cached KVs from Decode

        D->>D: Decode other requests
        P->>P: Run prefill
        P-->>D: Write prefilled KVs into allocated blocks
        P->>D: Send completion notification
        Note over D: Notification received when prefill is done
        D->>D: Schedule decoding
```

59
60
61
62
63
64
65
66
## Getting Started

1. Choose a deployment architecture based on your requirements
2. Configure the components as needed
3. Deploy using the provided scripts

### Prerequisites

67
Start required services (etcd and NATS) using [Docker Compose](../../deploy/docker-compose.yml)
68
69
70
71
72
```bash
docker compose -f deploy/docker-compose.yml up -d
```

### Build docker
73
74
75
76
77

```
./container/build.sh
```

78
### Run container
79
80
81
82

```
./container/run.sh -it
```
83
## Run Deployment
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

This figure shows an overview of the major components to deploy:

```
                                                 +----------------+
                                          +------| prefill worker |-------+
                                   notify |      |                |       |
                                 finished |      +----------------+       | pull
                                          v                               v
+------+      +-----------+      +------------------+    push     +---------------+
| HTTP |----->| processor |----->| decode/monolith  |------------>| prefill queue |
|      |<-----|           |<-----|      worker      |             |               |
+------+      +-----------+      +------------------+             +---------------+
                  |    ^                  |
       query best |    | return           | publish kv events
           worker |    | worker_id        v
                  |    |         +------------------+
                  |    +---------|     kv-router    |
                  +------------->|                  |
                                 +------------------+

```

107
### Example architectures
108
_Note_: For a non-dockerized deployment, first export `DYNAMO_HOME` to point to the dynamo repository root, e.g. `export DYNAMO_HOME=$(pwd)`
109

110
#### Aggregated serving
111
```bash
112
cd $DYNAMO_HOME/examples/llm
113
dynamo serve graphs.agg:Frontend -f ./configs/agg.yaml
114
115
```

116
#### Aggregated serving with KV Routing
117
```bash
118
cd $DYNAMO_HOME/examples/llm
119
dynamo serve graphs.agg_router:Frontend -f ./configs/agg_router.yaml
120
```
121

122
#### Disaggregated serving
123
```bash
124
cd $DYNAMO_HOME/examples/llm
125
dynamo serve graphs.disagg:Frontend -f ./configs/disagg.yaml
126
127
```

128
#### Disaggregated serving with KV Routing
129
```bash
130
cd $DYNAMO_HOME/examples/llm
131
dynamo serve graphs.disagg_router:Frontend -f ./configs/disagg_router.yaml
132
```
133

134
135
136
137
138
139
### Client

In another terminal:
```bash
# this test request has around 200 tokens isl

140
curl localhost:8000/v1/chat/completions   -H "Content-Type: application/json"   -d '{
141
142
143
144
145
146
147
148
149
150
    "model": "deepseek-ai/DeepSeek-R1-Distill-Llama-8B",
    "messages": [
    {
        "role": "user",
        "content": "In the heart of Eldoria, an ancient land of boundless magic and mysterious creatures, lies the long-forgotten city of Aeloria. Once a beacon of knowledge and power, Aeloria was buried beneath the shifting sands of time, lost to the world for centuries. You are an intrepid explorer, known for your unparalleled curiosity and courage, who has stumbled upon an ancient map hinting at ests that Aeloria holds a secret so profound that it has the potential to reshape the very fabric of reality. Your journey will take you through treacherous deserts, enchanted forests, and across perilous mountain ranges. Your Task: Character Background: Develop a detailed background for your character. Describe their motivations for seeking out Aeloria, their skills and weaknesses, and any personal connections to the ancient city or its legends. Are they driven by a quest for knowledge, a search for lost familt clue is hidden."
    }
    ],
    "stream":false,
    "max_tokens": 30
  }'
151

152
153
```

154
### Multi-node deployment
155

156
See [multinode-examples.md](multinode-examples.md) for more details.
157

158
159
### Close deployment

160
Kill all dynamo processes managed by circusd.
161
162

```
163
164
ctrl-c
pkill python3
165
```