From 5106816eace5e107c88f3f53729f13380c4cf440 Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Thu, 5 Dec 2024 17:05:20 +0800
Subject: [PATCH] drafted tokens debug prints

---
 gpttype_adapter.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/gpttype_adapter.cpp b/gpttype_adapter.cpp
index 11ab36126..0e17874db 100644
--- a/gpttype_adapter.cpp
+++ b/gpttype_adapter.cpp
@@ -3406,6 +3406,11 @@ generation_outputs gpttype_generate(const generation_inputs inputs)
                     draft_used = true;
                     draft_results = speculative_decoding_eval_chunk(draft_ctx, llama_ctx_v4, embd, n_vocab, n_past);
                     evalres = draft_results.draft_success;
+                    if(debugmode==1)
+                    {
+                        std::string draftedtoks = get_tok_vec_str(draft_results.draftids);
+                        printf("\nDrafted %d Tokens: [%s]\n",speculative_chunk_amt,draftedtoks.c_str());
+                    }
                 }
             }
             else if(file_format==FileFormat::RWKV_1 || file_format==FileFormat::RWKV_2)