huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 20:22:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Initial setup

Browse Source
This commit is contained in:
Ivar Flakstad
2023-10-31 18:09:10 +01:00
parent c12ad45562
commit 4525b7b52a
13 changed files with 519 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-pyo3/src/lib.rs
View File

@ -81,6 +81,7 @@ impl PyDevice {
match device {
Device::Cpu => Self::Cpu,
Device::Cuda(_) => Self::Cuda,
Device::Metal(_) => unimplemented!(),
}
}