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

Refactor to simplify our lives for settings the params in the encoder.

Browse Source
This commit is contained in:
Nicolas Patry
2023-11-10 01:24:49 +01:00
committed by Nicolas Patry
parent 39406a6721
commit df6814f34e
6 changed files with 339 additions and 255 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -146,6 +146,7 @@ impl Device {
match (self, rhs) {
(Self::Cpu, Self::Cpu) => true,
(Self::Cuda(lhs), Self::Cuda(rhs)) => lhs.same_device(rhs),
(Self::Metal(lhs), Self::Metal(rhs)) => lhs.same_device(rhs),
_ => false,
}
}