Clippy fix. (#1972)

This commit is contained in:
Laurent Mazare
2024-03-31 08:57:40 +02:00
committed by GitHub
parent 92f81d2fcb
commit eead1dcead

View File

@ -79,7 +79,7 @@ impl TextGeneration {
} else {
self.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 = if self.repeat_penalty == 1. {