huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 18:28:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use AsRef<str> for set_one. (#1033)

Browse Source
This commit is contained in:
Laurent Mazare
2023-10-05 06:05:44 +01:00
committed by GitHub
parent b86ac0c507
commit f0c619a4af
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-nn/src/var_map.rs
View File

@ -52,7 +52,7 @@ impl VarMap {
}
/// Set a named variable to some value.
pub fn set_one<K: AsRef<String>, V: AsRef<Tensor>>(&mut self, name: K, value: V) -> Result<()> {
pub fn set_one<K: AsRef<str>, V: AsRef<Tensor>>(&mut self, name: K, value: V) -> Result<()> {
let tensor_data = self.data.lock().unwrap();
let name = name.as_ref();
match tensor_data.get(name) {