Merge branch 'master' of github.com:DaniAsh551/rllama

Noeda · Apr 7, 2023 · 5b76ef4 · 5b76ef4
2 parents 3f35a84 + 8d6897e
commit 5b76ef4
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();