"git@developer.sourcefind.cn:tsoc/migraphx-auto-test.git" did not exist on "2e912f00c1e395a0c0026b160e791d31bd2cccd4"
Unverified Commit 37e0974a authored by Matt's avatar Matt Committed by GitHub
Browse files

Fix doctests for TFVisionTextDualEncoder (#21910)

parent 9f5bfe1b
......@@ -272,10 +272,10 @@ class TFVisionTextDualEncoderModel(TFPreTrainedModel):
```python
>>> from transformers import TFVisionTextDualEncoderModel, AutoTokenizer
>>> model = TFVisionTextDualEncoderModel.from_pretrained("clip-italian/clip-italian")
>>> model = TFVisionTextDualEncoderModel.from_pretrained("clip-italian/clip-italian", from_pt=True)
>>> tokenizer = AutoTokenizer.from_pretrained("clip-italian/clip-italian")
>>> inputs = tokenizer(["una foto di un gatto", "una foto di un cane"], padding=True, return_tensors="pt")
>>> inputs = tokenizer(["una foto di un gatto", "una foto di un cane"], padding=True, return_tensors="np")
>>> text_features = model.get_text_features(**inputs)
```"""
text_outputs = self.text_model(
......@@ -313,7 +313,7 @@ class TFVisionTextDualEncoderModel(TFPreTrainedModel):
>>> import requests
>>> from transformers import TFVisionTextDualEncoderModel, AutoImageProcessor
>>> model = VisionTextDualEncoderModel.from_pretrained("clip-italian/clip-italian")
>>> model = TFVisionTextDualEncoderModel.from_pretrained("clip-italian/clip-italian", from_pt=True)
>>> image_processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224")
>>> url = "http://images.cocodataset.org/val2017/000000039769.jpg"
......@@ -380,7 +380,7 @@ class TFVisionTextDualEncoderModel(TFPreTrainedModel):
... ]
>>> images = [Image.open(requests.get(url, stream=True).raw) for url in urls]
>>> inputs = processor(
... text=["a photo of a cat", "a photo of a dog"], images=images, return_tensors="pt", padding=True
... text=["a photo of a cat", "a photo of a dog"], images=images, return_tensors="np", padding=True
... )
>>> outputs = model(
... input_ids=inputs.input_ids,
......@@ -587,6 +587,8 @@ class TFVisionTextDualEncoderModel(TFPreTrainedModel):
if text_model.name != "text_model":
raise ValueError("text model must be created with the name `text_model`.")
model(model.dummy_inputs) # Ensure model is fully built
return model
@property
......
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