From 55e428c8aed766d7e2433acd032f9d2768facf5f Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Fri, 11 Aug 2023 17:58:56 +0200
Subject: [PATCH] Expose the varmap inner data. (#411)

---
 candle-nn/src/var_builder.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/candle-nn/src/var_builder.rs b/candle-nn/src/var_builder.rs
index 374260b0..311a6fb9 100644
--- a/candle-nn/src/var_builder.rs
+++ b/candle-nn/src/var_builder.rs
@@ -81,6 +81,10 @@ impl VarMap {
         tensor_data.insert(path.to_string(), var);
         Ok(tensor)
     }
+
+    pub fn data(&self) -> &Mutex<HashMap<String, Var>> {
+        &self.data
+    }
 }
 
 // TODO: Maybe we would want the storage to be generic, e.g. with Box<dyn> to avoid too many