Add the blip image captioning model (#1140)

* Blip text model.

* Blip vision bits.

* Blippity.

* More blip.
This commit is contained in:
Laurent Mazare
2023-10-20 22:09:11 +01:00
committed by GitHub
parent cfb423ab76
commit 34d9e91748
4 changed files with 595 additions and 2 deletions

View File

@ -58,8 +58,8 @@ pub struct Conv2d {
span: tracing::Span,
}
impl Conv2d {
pub fn forward(&self, x: &Tensor) -> Result<Tensor> {
impl Module for Conv2d {
fn forward(&self, x: &Tensor) -> Result<Tensor> {
let _enter = self.span.enter();
self.inner.forward(x)
}