From 97bc25a424b4e8d41a8f4c2b8302f625c1b94428 Mon Sep 17 00:00:00 2001
From: Laurent <laurent.mazare@gmail.com>
Date: Sun, 29 Sep 2024 19:49:32 +0200
Subject: [PATCH] Tweak the prompt.

---
 candle-examples/examples/paligemma/main.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/candle-examples/examples/paligemma/main.rs b/candle-examples/examples/paligemma/main.rs
index cdfe812c7..9ce5011bc 100644
--- a/candle-examples/examples/paligemma/main.rs
+++ b/candle-examples/examples/paligemma/main.rs
@@ -270,6 +270,7 @@ fn main() -> Result<()> {
         args.repeat_last_n,
         &device,
     );
-    pipeline.run(&args.prompt, args.sample_len)?;
+    let prompt = format!("{}\n", args.prompt);
+    pipeline.run(&prompt, args.sample_len)?;
     Ok(())
 }