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

[Proposal] Remove SafeTensor wrapper (allows finer control for users).

Browse Source
This commit is contained in:
Nicolas Patry
2023-07-19 16:25:44 +02:00
parent 67e20c3792
commit dfd624dbd3
4 changed files with 26 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-examples/examples/whisper/main.rs
View File

@ -10,7 +10,7 @@
extern crate intel_mkl_src;
use anyhow::{Error as E, Result};
use candle::{DType, Device, Tensor};
use candle::{safetensors::Load, DType, Device, Tensor};
use candle_hub::{api::sync::Api, Repo, RepoType};
use candle_nn::VarBuilder;
use clap::Parser;
@ -311,7 +311,7 @@ fn main() -> Result<()> {
let mel_filters = unsafe { candle::safetensors::MmapedFile::new(args.filters)? };
let mel_filters = mel_filters.deserialize()?;
let mel_filters = mel_filters.tensor("mel_80", &device)?;
let mel_filters = mel_filters.tensor("mel_80")?.load(&device)?;
println!("loaded mel filters {:?}", mel_filters.shape());
let mel_filters = mel_filters.flatten_all()?.to_vec1::<f32>()?;