diff --git a/common/log.h b/common/log.h index c0e814861e0c69..4b06ca24ba5b90 100644 --- a/common/log.h +++ b/common/log.h @@ -684,8 +684,12 @@ inline std::string LOG_BATCH_TOSTR_PRETTY(const C & ctx, const B & batch) << ":token '" << detokenized << "'" << ":pos " << std::to_string(batch.pos[i]) << ":n_seq_id " << std::to_string(batch.n_seq_id[i]) - << ":seq_id " << std::to_string(batch.seq_id[i][0]) - << ":logits " << std::to_string(batch.logits[i]); + << ":seq_id "; + for (int j = 0; j < batch.n_seq_id[i]; j++) { + buf << std::to_string(batch.seq_id[i][j]) << ", "; + } + + buf << ":logits " << std::to_string(batch.logits[i]); } buf << " ]";