summaryrefslogtreecommitdiff
path: root/candle-examples/examples/yolo-v3/main.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-examples/examples/yolo-v3/main.rs')
-rw-r--r--candle-examples/examples/yolo-v3/main.rs4
1 files changed, 1 insertions, 3 deletions
diff --git a/candle-examples/examples/yolo-v3/main.rs b/candle-examples/examples/yolo-v3/main.rs
index ecf75bdf..5b1937ac 100644
--- a/candle-examples/examples/yolo-v3/main.rs
+++ b/candle-examples/examples/yolo-v3/main.rs
@@ -146,9 +146,7 @@ pub fn main() -> Result<()> {
// Create the model and load the weights from the file.
let model = args.model()?;
- let weights = unsafe { candle::safetensors::MmapedFile::new(model)? };
- let weights = weights.deserialize()?;
- let vb = VarBuilder::from_safetensors(vec![weights], DType::F32, &Device::Cpu);
+ let vb = unsafe { VarBuilder::from_mmaped_safetensors(&[model], DType::F32, &Device::Cpu)? };
let config = args.config()?;
let darknet = darknet::parse_config(config)?;
let model = darknet.build_model(vb)?;