From 6cc5411d816f84359b3ec3440dcb61ed729cfb2d Mon Sep 17 00:00:00 2001 From: Raushan Turganbay Date: Tue, 12 Mar 2024 15:16:21 +0500 Subject: [PATCH] Fix Fuyu doc typos (#29601) fix fuyu docs --- docs/source/en/model_doc/fuyu.md | 2 +- src/transformers/models/fuyu/modeling_fuyu.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/en/model_doc/fuyu.md b/docs/source/en/model_doc/fuyu.md index 2832e35398..a2e7be90aa 100644 --- a/docs/source/en/model_doc/fuyu.md +++ b/docs/source/en/model_doc/fuyu.md @@ -81,7 +81,7 @@ text_prompt = "Generate a coco-style caption.\\n" bus_image_url = "https://huggingface.co/datasets/hf-internal-testing/fixtures-captioning/resolve/main/bus.png" bus_image_pil = Image.open(io.BytesIO(requests.get(bus_image_url).content)) -inputs_to_model = processor(text=text_prompt, images=image_pil) +inputs_to_model = processor(text=text_prompt, images=bus_image_pil) ``` diff --git a/src/transformers/models/fuyu/modeling_fuyu.py b/src/transformers/models/fuyu/modeling_fuyu.py index 0d2a121edd..94d9704631 100644 --- a/src/transformers/models/fuyu/modeling_fuyu.py +++ b/src/transformers/models/fuyu/modeling_fuyu.py @@ -246,7 +246,7 @@ class FuyuForCausalLM(FuyuPreTrainedModel): >>> image = Image.open(requests.get(url, stream=True).raw) >>> prompt = "Generate a coco-style caption.\n" - >>> inputs = processor(text=text_prompt, images=image, return_tensors="pt") + >>> inputs = processor(text=prompt, images=image, return_tensors="pt") >>> outputs = model(**inputs) >>> generated_ids = model.generate(**model_inputs, max_new_tokens=7)