fix: cuda serialization
parent
0226a379de
commit
277528d01c
|
|
@ -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