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

Fix linspace implementation (#1358)

Browse Source 
* Fix linspace implementation

`steps` should be strictly greater than 1 to make it consistent with the context.

* Handle steps == 0 and steps == 1.

* Fix rustfmt.

---------

Co-authored-by: laurent <laurent.mazare@gmail.com>
This commit is contained in:
MilkFather
2023-11-23 15:35:13 +08:00
committed by GitHub
parent ec23427d60
commit ca19a9af62
2 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-examples/examples/yolo-v8/main.rs
View File

@ -7,7 +7,7 @@ extern crate accelerate_src;
mod model; mod model;
use model::{Multiples, YoloV8, YoloV8Pose}; use model::{Multiples, YoloV8, YoloV8Pose};
use candle::{DType, IndexOp, Result, Tensor, Device}; use candle::{DType, Device, IndexOp, Result, Tensor};
use candle_nn::{Module, VarBuilder}; use candle_nn::{Module, VarBuilder};
use candle_transformers::object_detection::{non_maximum_suppression, Bbox, KeyPoint}; use candle_transformers::object_detection::{non_maximum_suppression, Bbox, KeyPoint};
use clap::{Parser, ValueEnum}; use clap::{Parser, ValueEnum};

9
candle-transformers/src/models/stable_diffusion/utils.rs
View File

@ -1,12 +1,15 @@
use candle::{Device, Result, Tensor}; use candle::{Device, Result, Tensor};
pub fn linspace(start: f64, stop: f64, steps: usize) -> Result<Tensor> { pub fn linspace(start: f64, stop: f64, steps: usize) -> Result<Tensor> {
if steps < 1 { if steps == 0 {
candle::bail!("cannot use linspace with steps {steps} <= 1") Tensor::from_vec(Vec::<f64>::new(), steps, &Device::Cpu)
} } else if steps == 1 {
Tensor::from_vec(vec![start], steps, &Device::Cpu)
} else {
let delta = (stop - start) / (steps - 1) as f64; let delta = (stop - start) / (steps - 1) as f64;
let vs = (0..steps) let vs = (0..steps)
.map(|step| start + step as f64 * delta) .map(|step| start + step as f64 * delta)
.collect::<Vec<_>>(); .collect::<Vec<_>>();
Tensor::from_vec(vs, steps, &Device::Cpu) Tensor::from_vec(vs, steps, &Device::Cpu)
}
} }