dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

This commit is contained in:
ydshieh 2024-04-12 17:23:31 +02:00
parent 332f2691d6
commit c5abac66cc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/fuyu/modeling_fuyu.py
View File

@ -250,7 +250,7 @@ class FuyuForCausalLM(FuyuPreTrainedModel):
>>> generated_ids = model.generate(**inputs, max_new_tokens=7)
>>> generation_text = processor.batch_decode(generated_ids[:, -7:], skip_special_tokens=True)
>>> print(generation_text[0])
'A blue 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