From dcfa2700f621235d549f4bf444493753c3af11b4 Mon Sep 17 00:00:00 2001
From: dc-dc-dc <dgcruz983@gmail.com>
Date: Mon, 18 Dec 2023 20:12:44 -0500
Subject: [PATCH] use the correct type

---
 mlx/safetensor.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlx/safetensor.cpp b/mlx/safetensor.cpp
index 7aca0c691..30f5d2b5a 100644
--- a/mlx/safetensor.cpp
+++ b/mlx/safetensor.cpp
@@ -306,7 +306,7 @@ std::unordered_map<std::string, array> load_safetensor(
     Dtype type = dtype_from_safetensor_str(dtype);
     auto loaded_array = array(
         shape_vec,
-        float32,
+        type,
         std::make_unique<Load>(
             to_stream(s),
             in_stream,