Distributed Inference: Switched the bonus section and fixed typos

content/learning-paths/servers-and-cloud-computing/distributed-inference-with-llama-cpp/_index.md

@@ -1,9 +1,9 @@
 ---
 title: Distributed inference using llama.cpp
 
-draft: true
-cascade:
-    draft: true
+# draft: true
+# cascade:
+#     draft: true
 
 minutes_to_complete: 30
 

content/learning-paths/servers-and-cloud-computing/distributed-inference-with-llama-cpp/how-to-1.md

@@ -1,64 +1,132 @@
 ---
-title: Overview and Worker Node Configuration
+title: Convert model to gguf and quantization (Optional)
 weight: 2
 
 ### FIXED, DO NOT MODIFY
 layout: learningpathall
 ---
+## Assumptions
+1.	This page assumes that Python is installed on your AWS (or other cloud provider) instance.
+2.	You have access to Meta’s gated repository for the Llama 3.1 model family.
+3.	You have a Hugging Face token generated for downloading models.
+4.	You are performing this download on the master node.
+5. You have cloned llama.cpp repository.
+6. Basic libraries such as: transformers, torch, sentencepiece are already installed.
+7. There is 2TB+ storage space on your device.
+8. The model conversion was performed on c8g.24xlarge. Please make sure you have an equivalent ma hine.
 
-## Before you begin
-The instructions in this Learning Path are for any Arm server running Ubuntu 24.04.2 LTS. You will need at least three Arm server instances with at least 64 cores and 128GB of RAM to run this example. The instructions have been tested on an AWS Graviton4 c8g.16xlarge instance
+{{% notice Note %}}The time mentioned on the Introduction page is only for setting up distributed inference (process on the next page). The gguf conversion mentioned on this page may take 6+ hours. You may skip this page, if you have a GGUF model of your choice. Before skipping please read the paragraph preceding: "Step 1: Make a virtual environment." {{% /notice %}}
 
-## Overview
-llama.cpp is a C++ library that enables efficient inference of LLaMA and similar large language models on CPUs, optimized for local and embedded environments. Just over a year ago from its publication date, rgerganov’s RPC code was merged into llama.cpp, enabling distributed inference of large LLMs across multiple CPU-based machines—even when the models don’t fit into the memory of a single machine. In this learning path, we’ll explore how to run a 405B parameter model on Arm-based CPUs.
-
-For the purposes of this demonstration, the following experimental setup will be used:
-- Total number of instances: 3
-- Instance type: c8g.16xlarge
-- Model: Llama-3.1-405B_Q4_0.gguf
-
-One of the three nodes will serve as the master node, which physically hosts the model file. The other two nodes will act as worker nodes. In llama.cpp, remote procedure calls (RPC) are used to offload both the model and the computation over TCP connections between nodes. The master node forwards inference requests to the worker nodes, where all the actual computation is performed.
-
-## Implementation
-
-1. To get started, follow [this learning path](/learning-paths/servers-and-cloud-computing/llama-cpu) up to the step where you clone the llama.cpp repository. Since this setup involves multiple instances (or devices), you will need to replicate the initial setup on each device. Specifically, after executing the command below on all devices, continue with this learning path starting from Step 2.
+## Procedure
+Implement the following steps to download, convert and quantize the 405B parameter model released my meta [here](https://huggingface.co/meta-llama/Llama-3.1-405B).
+To get started, follow [this learning path](/learning-paths/servers-and-cloud-computing/llama-cpu) up to the step where you clone the llama.cpp repository. Since this setup involves multiple instances (or devices), you will need to replicate the initial setup on each device. Specifically, after executing the command below on all devices, continue with this learning path starting from Step 2.
 
 ```bash
 git clone https://github.com/ggerganov/llama.cpp
 ```
-2. Now we can build the llama.cpp library with the RPC feature enabled by compiling it with the -DLLAMA_RPC=ON flag
+##### 1. Make a virtual environment
+Make a python virtual environment using (update the python version accordingly in the following commands): 
 ```bash
-cd llama.cpp
-mkdir -p build-rpc
-cd build-rpc
-cmake .. -DGGML_RPC=ON -DLLAMA_BUILD_SERVER=ON
-cmake --build . --config Release
+apt update
+apt install python3.12-venv
+python3 -m venv myenv
+source myenv/bin/activate
 ```
+##### 2. Download the model
+Install Huggingface Hub in the virtual environment: 
+```bash
+pip3 install huggingface_hub 
 
-`llama.cpp` is now built in the `build-rpc/bin` directory.
-Check that `llama.cpp` has built correctly by running the help command:
+```
+Make a python file and name it download.py:
 ```bash
-cd build-rpc
-bin/llama-cli -h
+vi download.py
 ```
-If everything was built correctly, you should see a list of all the available flags that can be used with llama-cli.
-3. Now, choose two of the three devices to act as backend workers. If the devices had varying compute capacities, the ones with the highest compute should be selected—especially for a 405B model. However, since all three devices have identical compute capabilities in this case, you can select any two to serve as backend workers.
-
-Communication between the master node and the worker nodes occurs through a socket created on each worker. This socket listens for incoming data from the master—such as model parameters, tokens, hidden states, and other inference-related information.
-{{% notice Note %}}The RPC feature in llama.cpp is not secure by default, so you should never expose it to the open internet. To mitigate this risk, ensure that the security groups for all your EC2 instances are properly configured—restricting access to only trusted IPs or internal VPC traffic. This helps prevent unauthorized access to the RPC endpoints.{{% /notice %}}
-Use the following command to start the listening on the worker nodes:
+Write the following code to it: 
+```python
+import os
+from huggingface_hub import snapshot_download
+model_id = "meta-llama/Llama-3.1-405B"
+local_dir = "llama-hf"
+# Create the directory if it doesn't exist
+os.makedirs(local_dir, exist_ok=True)
+# Download the model snapshot
+snapshot_download( repo_id=model_id, local_dir=local_dir,
+    revision="main",
+    token="your_hf_token",
+    allow_patterns=["*.md", "*.json", "*.safetensors"]
+)
+```
+Execute the file: 
 ```bash
-bin/rpc-server -p 50052 -H 0.0.0.0 -t 64
+python3 download.py
+```
+##### 3. Convert the model from .safetensors to gguf and quantize
+Following lines installs the files important for conversion to .gguf format.
+```bash
+pip3 install -r llama.cpp/requirements.txt
+python3 llama.cpp/convert_hf_to_gguf.py llama-hf
+cd llama.cpp/build-rpc
+bin/llama-quantize ../../llama-hf/Llama-Hf-406B-F16.gguf Q4_0
+```
+You may rename the resultant file to model.gguf and use it. There are different quantization options as well, as shown below: 
+```bash
+bin/llama-quantize -h
 ```
-Below are the available flag options that can be used with the rpc-server functionality:
-
 ```output
--h, --help                show this help message and exit
--t,      --threads        number of threads for the CPU backend (default: 6)
--d DEV,  --device         device to use
--H HOST, --host HOST      host to bind to (default: 127.0.0.1)
--p PORT, --port PORT      port to bind to (default: 50052)
--m MEM,  --mem MEM        backend memory size (in MB)
--c,      --cache          enable local file cache
+usage: bin/llama-quantize [--help] [--allow-requantize] [--leave-output-tensor] [--pure] [--imatrix] [--include-weights] [--exclude-weights] [--output-tensor-type]
+       [--token-embedding-type] [--tensor-type] [--keep-split] [--override-kv] model-f32.gguf [model-quant.gguf] type [nthreads]
+
+  --allow-requantize: Allows requantizing tensors that have already been quantized. Warning: This can severely reduce quality compared to quantizing from 16bit or 32bit
+  --leave-output-tensor: Will leave output.weight un(re)quantized. Increases model size but may also increase quality, especially when requantizing
+  --pure: Disable k-quant mixtures and quantize all tensors to the same type
+  --imatrix file_name: use data in file_name as importance matrix for quant optimizations
+  --include-weights tensor_name: use importance matrix for this/these tensor(s)
+  --exclude-weights tensor_name: use importance matrix for this/these tensor(s)
+  --output-tensor-type ggml_type: use this ggml_type for the output.weight tensor
+  --token-embedding-type ggml_type: use this ggml_type for the token embeddings tensor
+  --tensor-type TENSOR=TYPE: quantize this tensor to this ggml_type. example: --tensor-type attn_q=q8_0
+      Advanced option to selectively quantize tensors. May be specified multiple times.
+  --keep-split: will generate quantized model in the same shards as input
+  --override-kv KEY=TYPE:VALUE
+      Advanced option to override model metadata by key in the quantized model. May be specified multiple times.
+Note: --include-weights and --exclude-weights cannot be used together
+
+Allowed quantization types:
+   2  or  Q4_0    :  4.34G, +0.4685 ppl @ Llama-3-8B
+   3  or  Q4_1    :  4.78G, +0.4511 ppl @ Llama-3-8B
+   8  or  Q5_0    :  5.21G, +0.1316 ppl @ Llama-3-8B
+   9  or  Q5_1    :  5.65G, +0.1062 ppl @ Llama-3-8B
+  19  or  IQ2_XXS :  2.06 bpw quantization
+  20  or  IQ2_XS  :  2.31 bpw quantization
+  28  or  IQ2_S   :  2.5  bpw quantization
+  29  or  IQ2_M   :  2.7  bpw quantization
+  24  or  IQ1_S   :  1.56 bpw quantization
+  31  or  IQ1_M   :  1.75 bpw quantization
+  36  or  TQ1_0   :  1.69 bpw ternarization
+  37  or  TQ2_0   :  2.06 bpw ternarization
+  10  or  Q2_K    :  2.96G, +3.5199 ppl @ Llama-3-8B
+  21  or  Q2_K_S  :  2.96G, +3.1836 ppl @ Llama-3-8B
+  23  or  IQ3_XXS :  3.06 bpw quantization
+  26  or  IQ3_S   :  3.44 bpw quantization
+  27  or  IQ3_M   :  3.66 bpw quantization mix
+  12  or  Q3_K    : alias for Q3_K_M
+  22  or  IQ3_XS  :  3.3 bpw quantization
+  11  or  Q3_K_S  :  3.41G, +1.6321 ppl @ Llama-3-8B
+  12  or  Q3_K_M  :  3.74G, +0.6569 ppl @ Llama-3-8B
+  13  or  Q3_K_L  :  4.03G, +0.5562 ppl @ Llama-3-8B
+  25  or  IQ4_NL  :  4.50 bpw non-linear quantization
+  30  or  IQ4_XS  :  4.25 bpw non-linear quantization
+  15  or  Q4_K    : alias for Q4_K_M
+  14  or  Q4_K_S  :  4.37G, +0.2689 ppl @ Llama-3-8B
+  15  or  Q4_K_M  :  4.58G, +0.1754 ppl @ Llama-3-8B
+  17  or  Q5_K    : alias for Q5_K_M
+  16  or  Q5_K_S  :  5.21G, +0.1049 ppl @ Llama-3-8B
+  17  or  Q5_K_M  :  5.33G, +0.0569 ppl @ Llama-3-8B
+  18  or  Q6_K    :  6.14G, +0.0217 ppl @ Llama-3-8B
+   7  or  Q8_0    :  7.96G, +0.0026 ppl @ Llama-3-8B
+   1  or  F16     : 14.00G, +0.0020 ppl @ Mistral-7B
+  32  or  BF16    : 14.00G, -0.0050 ppl @ Mistral-7B
+   0  or  F32     : 26.00G              @ 7B
+          COPY    : only copy tensors, no quantizing
 ```
-Setting the host to 0.0.0.0 might seem counterintuitive given the earlier security warning, but it’s acceptable in this case because the security groups have been properly configured to block any unintended or unauthorized access.