Update pickle loading fn

EricLBuehler · Feb 6, 2024 · 232d8bd · 232d8bd
1 parent 7fed136
commit 232d8bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/candle-lora-transformers/src/varbuilder_utils.rs b/candle-lora-transformers/src/varbuilder_utils.rs
@@ -95,7 +95,7 @@ pub fn from_pth_tensors<'a, P: AsRef<Path>>(
     {
         let mut ws = map.data().lock().unwrap();
 
-        let tensors = candle_core::pickle::PthTensors::new(path)?;
+        let tensors = candle_core::pickle::PthTensors::new(path, None)?;
         if silent {
             for name in tensors.tensor_infos().keys() {
                 let tensor = tensors