summaryrefslogtreecommitdiff
path: root/examples
diff options
context:
space:
mode:
authorHanishKVC <hanishkvc@gmail.com>2024-05-10 15:51:58 +0530
committerGitHub <noreply@github.com>2024-05-10 20:21:58 +1000
commitf89fe2732c5709f6e86d5f4aee2e6d2a561f2eb2 (patch)
tree4e427e6595de0d43d601c7781c16445297e5322f /examples
parentd11afd665241c1b3910ab5f040d0216403019d87 (diff)
Main+: optionally allow special tokens from user in interactive mode (#7097)
@hanishkvc added a new `--interactive-specials` flag which would allow for inserting special tokens from user side into the embedding stream.
Diffstat (limited to 'examples')
-rw-r--r--examples/main/main.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/main/main.cpp b/examples/main/main.cpp
index 49acd6ba..f3e445c1 100644
--- a/examples/main/main.cpp
+++ b/examples/main/main.cpp
@@ -879,7 +879,7 @@ int main(int argc, char ** argv) {
}
const auto line_pfx = ::llama_tokenize(ctx, params.input_prefix, false, true);
- const auto line_inp = ::llama_tokenize(ctx, buffer, false, false);
+ const auto line_inp = ::llama_tokenize(ctx, buffer, false, params.interactive_specials);
const auto line_sfx = ::llama_tokenize(ctx, params.input_suffix, false, true);
LOG("input tokens: %s\n", LOG_TOKENS_TOSTR_PRETTY(ctx, line_inp).c_str());