From 042428ef1406681dd4b74ed20a0fe8f17699b0e6 Mon Sep 17 00:00:00 2001 From: kaixuanliu Date: Mon, 16 Dec 2024 07:55:31 +0000 Subject: [PATCH] change default dtype to bfloat16 Signed-off-by: kaixuanliu --- 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 } } }