huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 19:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix a couple typos (#1451)

Browse Source 
* Mixtral quantized instruct.

* Fix a couple typos.
This commit is contained in:
Laurent Mazare
2023-12-17 05:20:05 -06:00
committed by GitHub
parent c630622a07
commit 1e86717bf2
24 changed files with 44 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-core/src/tensor.rs
View File

@ -1,4 +1,4 @@
//! Tensors are N-dimenional matrixes of elements using a single data type.
//! Tensors are N-dimensional matrixes of elements using a single data type.
#![allow(clippy::redundant_closure_call)]
use crate::backend::{BackendDevice, BackendStorage};
use crate::op::{
@ -669,7 +669,7 @@ impl Tensor {
}
/// Split a tensor into the specified number of chunks, this may return less chunks than
/// specificed.
/// specified.
pub fn chunk<D: Dim>(&self, chunks: usize, dim: D) -> Result<Vec<Self>> {
let dim = dim.to_index(self.shape(), "chunk")?;
let size = self.dim(dim)?;