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