Tweak the prompt.

huggingface · Sep 29, 2024 · 97bc25a · 97bc25a
1 parent 2999594
commit 97bc25a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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(())
 }