From b2e8a2e22fbae9903b949a3e1204feda87fecc18 Mon Sep 17 00:00:00 2001
From: Georgi Gerganov <ggerganov@gmail.com>
Date: Wed, 6 Dec 2023 10:04:49 +0200
Subject: [PATCH] minor : add braces

---
 examples/speculative/speculative.cpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/examples/speculative/speculative.cpp b/examples/speculative/speculative.cpp
index f1d2e6eeb..dca3f84a5 100644
--- a/examples/speculative/speculative.cpp
+++ b/examples/speculative/speculative.cpp
@@ -203,8 +203,9 @@ int main(int argc, char ** argv) {
 
             const std::string token_str = llama_token_to_piece(ctx_tgt, id);
 
-            if (!params.use_color)
+            if (!params.use_color) {
                 printf("%s", token_str.c_str());
+            }
 
             if (id == llama_token_eos(model_tgt)) {
                 has_eos = true;
@@ -244,8 +245,9 @@ int main(int argc, char ** argv) {
                     continue;
                 }
             }
-            if (params.use_color)
+            if (params.use_color) {
                 printf("%s", token_str.c_str());
+            }
             fflush(stdout);
 
             LOG("the sampled target token (%d, '%s') did not match, or we ran out of drafted tokens\n", id, token_str.c_str());