From 621434e424d5f35b2913d867ddd7781b1c6cb1ad Mon Sep 17 00:00:00 2001
From: Naoki Kishida <naokikishida@gmail.com>
Date: Sun, 25 Feb 2024 07:21:01 +0900
Subject: [PATCH] reset conversation (#34)

---
 run.cc | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/run.cc b/run.cc
index 87d8445..96ba316 100644
--- a/run.cc
+++ b/run.cc
@@ -144,6 +144,11 @@ void ReplGemma(gcpp::Gemma& model, hwy::ThreadPool& pool,
       return;
     }
 
+    if (prompt_string == "%c" || prompt_string == "%C") {
+      abs_pos = 0;
+      continue;
+    }
+
     if (model.model_training == ModelTraining::GEMMA_IT) {
       // For instruction-tuned models: add control tokens.
       prompt_string = "<start_of_turn>user\n" + prompt_string +