Skip to content

Commit 4ad0676

Browse files
committed
parallel : fix crash when -n -1
1 parent 2585690 commit 4ad0676

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

examples/parallel/parallel.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ int main(int argc, char ** argv) {
114114
for (size_t i = 0; i < clients.size(); ++i) {
115115
auto & client = clients[i];
116116
client.id = i;
117-
client.tokens_prev.resize(params.n_predict);
117+
client.tokens_prev.resize(std::max(256, params.n_predict));
118118
std::fill(client.tokens_prev.begin(), client.tokens_prev.end(), 0);
119119
}
120120

@@ -321,7 +321,8 @@ int main(int argc, char ** argv) {
321321
// client.id, client.seq_id, id, client.n_decoded, client.i_batch, token_str.c_str());
322322

323323
if (client.n_decoded > 2 &&
324-
(id == llama_token_eos(ctx) || client.n_decoded + client.n_prompt >= params.n_predict ||
324+
(id == llama_token_eos(ctx) ||
325+
(params.n_predict > 0 && client.n_decoded + client.n_prompt >= params.n_predict) ||
325326
client.response.find("User:") != std::string::npos ||
326327
client.response.find('\n') != std::string::npos)) {
327328
// basic reverse prompt

0 commit comments

Comments
 (0)