From 4a6bdc12598a4e86122485dc604077e777c0a817 Mon Sep 17 00:00:00 2001 From: "Liu, Kaixuan" Date: Mon, 16 Dec 2024 12:02:09 -0500 Subject: [PATCH] change default dtype to BFloat16 Signed-off-by: Liu, Kaixuan --- backends/src/dtype.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backends/src/dtype.rs b/backends/src/dtype.rs index 8cc02ff8..b303fd2b 100644 --- a/backends/src/dtype.rs +++ b/backends/src/dtype.rs @@ -52,7 +52,7 @@ impl Default for DType { feature = "ort" )))] { - DType::Float16 + DType::BFloat16 } } }