fix: cuda serialization
parent
cd8edc2fef
commit
3c3b14c9f5
|
|
@ -76,6 +76,7 @@ pub enum Device {
|
||||||
Cpu,
|
Cpu,
|
||||||
|
|
||||||
#[cfg(feature = "cuda")]
|
#[cfg(feature = "cuda")]
|
||||||
|
#[strum(serialize = "cuda")]
|
||||||
Cuda,
|
Cuda,
|
||||||
|
|
||||||
#[cfg(all(target_os = "macos", target_arch = "aarch64"))]
|
#[cfg(all(target_os = "macos", target_arch = "aarch64"))]
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue