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

Transpose the weight matrixes for llama2.c. (#321)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-04 13:32:20 +01:00
committed by GitHub
parent 8b6f5be1cc
commit f7b2a0391d
2 changed files with 22 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
candle-core/src/device.rs
View File

@ -101,6 +101,13 @@ impl Device {
}
}
pub fn is_cpu(&self) -> bool {
match self {
Self::Cpu => true,
Self::Cuda(_) => false,
}
}
pub fn is_cuda(&self) -> bool {
match self {
Self::Cpu => false,