Log remaining prompt.

Ai00-X · Jul 26, 2023 · 8e85972 · 8e85972
1 parent 6c7937b
commit 8e85972
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -222,7 +222,7 @@ fn model_task(
             token_counter.prompt_tokens = tokens.len();
             token_counter.total_tokens = tokens.len();
 
-            log::trace!("{prompt}");
+            log::trace!("{}", String::from_utf8(remain).unwrap_or_default());
 
             for _ in 0..max_tokens {
                 if token_sender.is_disconnected() {
@@ -267,7 +267,7 @@ fn model_task(
             let _ = token_sender.send(Token::Stop(FinishReason::Length, token_counter));
         }
 
-        print!("\n\n");
+        println!("[DONE]");
 
         if let Ok(back) = state.back() {
             if embedding {