Log all sequence IDs in a batch

AutonomicPerfectionist · Jan 10, 2024 · 0082332 · 0082332
1 parent d23ab1d
commit 0082332
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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 << " ]";