fix: formatting in candle-examples/examples/

huggingface · Oct 6, 2024 · 5f0741d · 5f0741d
1 parent 714b72e
commit 5f0741d
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 12 deletions.
diff --git a/candle-examples/examples/codegeex4-9b/main.rs b/candle-examples/examples/codegeex4-9b/main.rs
@@ -102,9 +102,7 @@ impl TextGeneration {
  .decode(&[next_token], true)
  .expect("Token error");
  if self.verbose_prompt {
- println!(
- "[Count: {count}] [Raw Token: {next_token}] [Decode Token: {token}]"
- );
+ println!("[Count: {count}] [Raw Token: {next_token}] [Decode Token: {token}]");
  }
  result.push(token);
  std::io::stdout().flush()?;

diff --git a/candle-examples/examples/glm4/main.rs b/candle-examples/examples/glm4/main.rs
@@ -107,9 +107,7 @@ impl TextGeneration {
  .decode(&[next_token], true)
  .expect("Token error");
  if self.verbose_prompt {
- println!(
- "[Count: {count}] [Raw Token: {next_token}] [Decode Token: {token}]"
- );
+ println!("[Count: {count}] [Raw Token: {next_token}] [Decode Token: {token}]");
  }
  result.push(token);
  std::io::stdout().flush()?;

diff --git a/candle-examples/examples/llava/main.rs b/candle-examples/examples/llava/main.rs
@@ -206,9 +206,8 @@ fn main() -> Result<()> {
  let llava: LLaVA = LLaVA::load(vb, &llava_config, clip_vision_config)?;
 
  println!("generating conv template");
- let image_token_se = format!(
- "{DEFAULT_IM_START_TOKEN}{DEFAULT_IMAGE_TOKEN}{DEFAULT_IM_END_TOKEN}"
- );
+ let image_token_se =
+ format!("{DEFAULT_IM_START_TOKEN}{DEFAULT_IMAGE_TOKEN}{DEFAULT_IM_END_TOKEN}");
  let qs = if args.prompt.contains(IMAGE_PLACEHOLDER) {
  if llava_config.mm_use_im_start_end {
  args.prompt.replace(IMAGE_PLACEHOLDER, &image_token_se)

diff --git a/candle-examples/examples/quantized-qwen2-instruct/main.rs b/candle-examples/examples/quantized-qwen2-instruct/main.rs
@@ -212,9 +212,7 @@ fn main() -> anyhow::Result<()> {
  let tokenizer = args.tokenizer()?;
  let mut tos = TokenOutputStream::new(tokenizer);
  let prompt_str = args.prompt.unwrap_or_else(|| DEFAULT_PROMPT.to_string());
- let prompt_str = format!(
- "<|im_start|>user\n{prompt_str}<|im_end|>\n<|im_start|>assistant\n"
- );
+ let prompt_str = format!("<|im_start|>user\n{prompt_str}<|im_end|>\n<|im_start|>assistant\n");
  print!("formatted instruct prompt: {}", &prompt_str);
  let tokens = tos
  .tokenizer()