Skip to content

Commit

Permalink
clippy fix
Browse files Browse the repository at this point in the history
  • Loading branch information
ameroyer committed Dec 23, 2024
1 parent 732994b commit 583b68a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions candle-transformers/src/models/pixtral/vision_model.rs
Original file line number Diff line number Diff line change
Expand Up @@ -342,8 +342,8 @@ impl Model {
let idx = Tensor::arange(0, num_patches_h as u32, device)?;
let idy = Tensor::arange(0, num_patches_w as u32, device)?;
let mesh = Tensor::meshgrid(&[idx, idy], false)?;
let ids = (&mesh[0] * (self.max_image_width as f64) + &mesh[1])?;
Ok(ids.flatten_all())
let ids = (&mesh[0] * (self.max_image_width as f64) + &mesh[1])?.flatten_all()?;
Ok(ids)
}
}

Expand Down

0 comments on commit 583b68a

Please sign in to comment.