File tree Expand file tree Collapse file tree 1 file changed +19
-1
lines changed Expand file tree Collapse file tree 1 file changed +19
-1
lines changed Original file line number Diff line number Diff line change @@ -1583,7 +1583,15 @@ int32_t llama_context::output_reserve(int32_t n_outputs) {
15831583 embd = has_embd ? output_base + logits_size : nullptr ;
15841584
15851585 // set all ids as invalid (negative)
1586- std::fill (output_ids.begin (), output_ids.end (), -1 );
1586+ // llama-server-one START
1587+ #ifndef COSMOCC
1588+ std::fill (output_ids.begin (), output_ids.end (), -1 );
1589+ #else
1590+ for (auto iii = output_ids.begin (); iii != output_ids.end (); iii++) {
1591+ *iii = -1 ;
1592+ }
1593+ #endif
1594+ // llama-server-one END
15871595
15881596 ggml_backend_buffer_clear (buf_output.get (), 0 );
15891597
@@ -1623,7 +1631,17 @@ void llama_context::output_reorder() {
16231631 }
16241632 }
16251633 }
1634+
1635+ // llama-server-one START
1636+ #ifndef COSMOCC
16261637 std::fill (output_ids.begin (), output_ids.end (), -1 );
1638+ #else
1639+ for (auto iii = output_ids.begin (); iii != output_ids.end (); iii++) {
1640+ *iii = -1 ;
1641+ }
1642+ #endif
1643+ // llama-server-one END
1644+
16271645 for (int32_t i = 0 ; i < n_outputs; ++i) {
16281646 output_ids[out_ids[i]] = i;
16291647 }
You can’t perform that action at this time.
0 commit comments