huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 19:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use is_some to check if seed is present

Browse Source
This commit is contained in:
Mateusz Okulus
2024-04-19 16:09:45 +02:00
parent 46073c5f73
commit 0fa41a791f
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
candle-onnx/src/eval.rs
View File

@ -850,11 +850,8 @@ pub fn simple_eval(
let low: f32 = get_attr_opt(node, "low")?.copied().unwrap_or(0.0); let low: f32 = get_attr_opt(node, "low")?.copied().unwrap_or(0.0);
let high: f32 = get_attr_opt(node, "high")?.copied().unwrap_or(1.0); let high: f32 = get_attr_opt(node, "high")?.copied().unwrap_or(1.0);
let seed: Option<f32> = get_attr_opt(node, "seed")?.copied(); let seed: Option<f32> = get_attr_opt(node, "seed")?.copied();
match seed { if seed.is_some() {
Some(_) => { bail!("seed for RandomUniform is currently not supported")
bail!("seed for RandomUniform is currently not supported")
}
None => {}
}; };
let shape: Vec<usize> = get_attr::<[i64]>(node, "shape")? let shape: Vec<usize> = get_attr::<[i64]>(node, "shape")?
.iter() .iter()