From cac81f2d6f933f1c938136357091f180329e5caf Mon Sep 17 00:00:00 2001
From: stormy/cli <chengcli@umich.edu>
Date: Sat, 17 Jan 2026 14:54:51 -0500
Subject: [PATCH] wip

---
 src/utils/serialize.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/utils/serialize.cpp b/src/utils/serialize.cpp
index ebd2748..60ca741 100644
--- a/src/utils/serialize.cpp
+++ b/src/utils/serialize.cpp
@@ -20,7 +20,11 @@ void load_tensors(std::map<std::string, torch::Tensor>& tensor_map,
   torch::serialize::InputArchive archive;
   archive.load_from(filename);
   for (auto& pair : tensor_map) {
-    archive.read(pair.first, pair.second);
+    try {
+      archive.read(pair.first, pair.second);
+    } catch (const c10::Error& e) {
+      // skip missing tensors
+    }
   }
 }