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

Fix for whisper example. rand::distribution is now rand::distr (#2811)

Browse Source
This commit is contained in:
Benjamin Beurdouche
2025-03-16 19:14:55 +01:00
committed by GitHub
parent 3afb04925a
commit 0b24f7f0a4
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -14,7 +14,9 @@ use candle::{Device, IndexOp, Tensor};
use candle_nn::{ops::softmax, VarBuilder}; use candle_nn::{ops::softmax, VarBuilder};
use clap::{Parser, ValueEnum}; use clap::{Parser, ValueEnum};
use hf_hub::{api::sync::Api, Repo, RepoType}; use hf_hub::{api::sync::Api, Repo, RepoType};
use rand::{distributions::Distribution, SeedableRng}; use rand::distr::weighted::WeightedIndex;
use rand::distr::Distribution;
use rand::SeedableRng;
use tokenizers::Tokenizer; use tokenizers::Tokenizer;
mod multilingual; mod multilingual;
@ -208,7 +210,7 @@ impl Decoder {
let next_token = if t > 0f64 { let next_token = if t > 0f64 {
let prs = softmax(&(&logits / t)?, 0)?; let prs = softmax(&(&logits / t)?, 0)?;
let logits_v: Vec<f32> = prs.to_vec1()?; let logits_v: Vec<f32> = prs.to_vec1()?;
let distr = rand::distributions::WeightedIndex::new(&logits_v)?; let distr = WeightedIndex::new(&logits_v)?;
distr.sample(&mut self.rng) as u32 distr.sample(&mut self.rng) as u32
} else { } else {
let logits_v: Vec<f32> = logits.to_vec1()?; let logits_v: Vec<f32> = logits.to_vec1()?;