From: Georgi Gerganov <redacted>
Date: Fri, 12 Jul 2024 11:48:04 +0000 (+0300)
Subject: main : print error on empty input (#8456)
X-Git-Tag: upstream/0.0.4488~1104
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=6af51c0d96e6268769fc05c98d5b1a5e832c0017;p=pkg%2Fggml%2Fsources%2Fllama.cpp

main : print error on empty input (#8456)
---

diff --git a/examples/main/main.cpp b/examples/main/main.cpp
index 4ef55c1e..a0d817b1 100644
--- a/examples/main/main.cpp
+++ b/examples/main/main.cpp
@@ -289,8 +289,13 @@ int main(int argc, char ** argv) {
 
     // Should not run without any tokens
     if (embd_inp.empty()) {
-        embd_inp.push_back(llama_token_bos(model));
-        LOG("embd_inp was considered empty and bos was added: %s\n", LOG_TOKENS_TOSTR_PRETTY(ctx, embd_inp).c_str());
+        if (add_bos) {
+            embd_inp.push_back(llama_token_bos(model));
+            LOG("embd_inp was considered empty and bos was added: %s\n", LOG_TOKENS_TOSTR_PRETTY(ctx, embd_inp).c_str());
+        } else {
+            LOG_TEE("error: input is empty\n");
+            return -1;
+        }
     }
 
     // Tokenize negative prompt