Correct typo in cuda device name for rust chat model (#1241)

mlc-ai · Nov 13, 2023 · 26fd019 · 26fd019
1 parent c2082d8
commit 26fd019
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rust/src/chat_module.rs b/rust/src/chat_module.rs
@@ -278,7 +278,7 @@ impl ChatModule {
 
         // 1. Get device name and id
         let device_type = match device_name {
-            "cude" => 2,
+            "cuda" => 2,
             "opencl" => 4,
             "vulkan" => 7,
             "metal" => 8,