Unverified Commit e36a83d3 authored by AK391's avatar AK391 Committed by GitHub
Browse files

Merge branch 'huggingface:master' into master

parents cac87742 cc406da4
......@@ -341,7 +341,7 @@ class VisionTextDualEncoderModel(PreTrainedModel):
... pixel_values=inputs.pixel_values,
... return_loss=True,
... )
>>> loss, logits_per_image = outputs.loss, outputs.logits_per_imag # this is the image-text similarity score
>>> loss, logits_per_image = outputs.loss, outputs.logits_per_image # this is the image-text similarity score
>>> # save and load from pretrained
>>> model.save_pretrained("vit-bert")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment