change default dtype to BFloat16

Signed-off-by: Liu, Kaixuan <[email protected]>
kaixuanliu · Dec 16, 2024 · 4a6bdc1 · 4a6bdc1
1 parent cec3186
commit 4a6bdc1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backends/src/dtype.rs b/backends/src/dtype.rs
@@ -52,7 +52,7 @@ impl Default for DType {
             feature = "ort"
         )))]
         {
-            DType::Float16
+            DType::BFloat16
         }
     }
 }