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

Automatically upcast for to_u64 (#2244)

Browse Source
This commit is contained in:
Eric Buehler
2024-06-04 05:32:36 -04:00
committed by GitHub
parent 3f13ad3d79
commit 9182c828e6
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
candle-core/src/quantized/gguf_file.rs
View File

@ -217,10 +217,16 @@ impl Value {
} }
} }
/// This will also automatically upcast any integral types which will not truncate.
pub fn to_u64(&self) -> Result<u64> { pub fn to_u64(&self) -> Result<u64> {
match self { match self {
Self::U64(v) => Ok(*v), Self::U64(v) => Ok(*v),
v => crate::bail!("not a u64 {v:?}"), // Autoupcast cases here
Self::U8(v) => Ok(*v as u64),
Self::U16(v) => Ok(*v as u64),
Self::U32(v) => Ok(*v as u64),
Self::Bool(v) => Ok(*v as u64),
v => crate::bail!("not a u64 or upcastable to u64 {v:?}"),
} }
} }