Don't report interactive stop token sequence if we are not interactiv…

…e mode.
Noeda · Apr 2, 2023 · 4c99e7f · 4c99e7f
1 parent f5f8547
commit 4c99e7f
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/rllama_main.rs b/src/rllama_main.rs
@@ -754,6 +754,12 @@ fn command_line_inference(
         "Repetition penalty: {}",
         token_sampler.get_repetition_penalty()
     );
+    if start_interactive {
+        pln!(
+            "  Interactive mode stop token sequence: {}",
+            interactive_stop.as_str()
+        );
+    }
     pln!("---");
     pln!(
         "{}",
@@ -763,10 +769,6 @@ fn command_line_inference(
         "{}",
         "  This is the color of the generated text".truecolor(128, 255, 128)
     );
-    pln!(
-        "Interactive mode stop token sequence: {}",
-        interactive_stop.as_str()
-    );
     pln!("---");
     print!("{}", prompt.as_str().truecolor(128, 128, 255));