Unverified Commit 65657d5d authored by Pablo Montalvo's avatar Pablo Montalvo Committed by GitHub
Browse files

fix fuyu doctest (#30215)



* fix doctest

* fix example

* fix

* fix

---------
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent ac33aeee
...@@ -242,17 +242,17 @@ class FuyuForCausalLM(FuyuPreTrainedModel): ...@@ -242,17 +242,17 @@ class FuyuForCausalLM(FuyuPreTrainedModel):
>>> processor = FuyuProcessor.from_pretrained("adept/fuyu-8b") >>> processor = FuyuProcessor.from_pretrained("adept/fuyu-8b")
>>> model = FuyuForCausalLM.from_pretrained("adept/fuyu-8b") >>> model = FuyuForCausalLM.from_pretrained("adept/fuyu-8b")
>>> url = "http://images.cocodataset.org/val2017/000000039769.jpg" >>> url = "https://huggingface.co/datasets/hf-internal-testing/fixtures-captioning/resolve/main/bus.png"
>>> image = Image.open(requests.get(url, stream=True).raw) >>> image = Image.open(requests.get(url, stream=True).raw)
>>> prompt = "Generate a coco-style caption.\n" >>> prompt = "Generate a coco-style caption.\n"
>>> inputs = processor(text=prompt, images=image, return_tensors="pt") >>> inputs = processor(text=prompt, images=image, return_tensors="pt")
>>> outputs = model(**inputs) >>> outputs = model(**inputs)
>>> generated_ids = model.generate(**model_inputs, max_new_tokens=7) >>> generated_ids = model.generate(**inputs, max_new_tokens=7)
>>> generation_text = processor.batch_decode(generated_ids, skip_special_tokens=True) >>> generation_text = processor.batch_decode(generated_ids[:, -7:], skip_special_tokens=True)
>>> print(generation_text) >>> print(generation_text[0])
'A bus parked on the side of a road.' A blue bus parked on the side of a road.
```""" ```"""
output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
......
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