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): ...@@ -341,7 +341,7 @@ class VisionTextDualEncoderModel(PreTrainedModel):
... pixel_values=inputs.pixel_values, ... pixel_values=inputs.pixel_values,
... return_loss=True, ... 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 >>> # save and load from pretrained
>>> model.save_pretrained("vit-bert") >>> 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