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

Clippy fix. (#1972)

Browse Source
This commit is contained in:
Laurent Mazare
2024-03-31 08:57:40 +02:00
committed by GitHub
parent 92f81d2fcb
commit eead1dcead
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-examples/examples/moondream/main.rs
View File

@ -79,7 +79,7 @@ impl TextGeneration {
} else { } else {
self.model self.model
.text_model .text_model
.forward_with_img(&input, &image_embeds)? .forward_with_img(&input, image_embeds)?
}; };
let logits = logits.squeeze(0)?.to_dtype(DType::F32)?; let logits = logits.squeeze(0)?.to_dtype(DType::F32)?;
let logits = if self.repeat_penalty == 1. { let logits = if self.repeat_penalty == 1. {