huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expose the varmap inner data. (#411)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-11 17:58:56 +02:00
committed by GitHub
parent 01ea57da8c
commit 55e428c8ae
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-nn/src/var_builder.rs
View File

@ -81,6 +81,10 @@ impl VarMap {
tensor_data.insert(path.to_string(), var); tensor_data.insert(path.to_string(), var);
Ok(tensor) 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 // TODO: Maybe we would want the storage to be generic, e.g. with Box<dyn> to avoid too many