diff options
Diffstat (limited to 'candle-examples/examples/yolo-v3/main.rs')
-rw-r--r-- | candle-examples/examples/yolo-v3/main.rs | 4 |
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)?; |