fix: cuda serialization
parent
cd8edc2fef
commit
3c3b14c9f5
|
|
@ -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