Fix interactive system prompt overwriting --prompt.

Oops.
Noeda · Apr 7, 2023 · 8d6897e · 8d6897e
1 parent 61b68c6
commit 8d6897e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/rllama_main.rs b/src/rllama_main.rs
@@ -743,7 +743,9 @@ fn command_line_inference(
                 .into(),
         );
     }
-    prompt = interactive_system_prompt.clone();
+    if start_interactive {
+        prompt = interactive_system_prompt.clone();
+    }
 
     let mut toks_id: Vec<TokenId> = tok.tokenize_to_ids(prompt.clone());
     let mut toks_str: String = prompt.clone();