From c043d276825c68f42ca0296bb355009017ac7b79 Mon Sep 17 00:00:00 2001
From: dc-dc-dc <dgcruz983@gmail.com>
Date: Tue, 19 Dec 2023 14:22:09 -0500
Subject: [PATCH] more cleanup

---
 mlx/ops.h          | 2 +-
 mlx/safetensor.cpp | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/mlx/ops.h b/mlx/ops.h
index d8bc05c72..c0feaa402 100644
--- a/mlx/ops.h
+++ b/mlx/ops.h
@@ -1057,7 +1057,7 @@ array dequantize(
     int bits = 4,
     StreamOrDevice s = {});
 
-/** Load array from .safetensor file format */
+/** Load array map from .safetensor file format */
 std::unordered_map<std::string, array> load_safetensor(
     std::shared_ptr<io::Reader> in_stream,
     StreamOrDevice s = {});
diff --git a/mlx/safetensor.cpp b/mlx/safetensor.cpp
index 1ec3f2964..37d7ceb22 100644
--- a/mlx/safetensor.cpp
+++ b/mlx/safetensor.cpp
@@ -85,7 +85,6 @@ std::unordered_map<std::string, array> load_safetensor(
   char rawJson[jsonHeaderLength];
   in_stream->read(rawJson, jsonHeaderLength);
   auto metadata = json::parse(rawJson, rawJson + jsonHeaderLength);
-  // auto metadata = io::jsonDeserialize(json, jsonHeaderLength);
   // Should always be an object on the top-level
   if (!metadata.is_object()) {
     throw std::runtime_error(