Unverified Commit 66b15efb authored by Joao Gante's avatar Joao Gante Committed by GitHub
Browse files

Pix2struct: doctest fix (#22761)

parent 390e121f
......@@ -1700,8 +1700,8 @@ class Pix2StructForConditionalGeneration(Pix2StructPreTrainedModel):
>>> # forward pass
>>> outputs = model(**inputs, labels=labels)
>>> loss = outputs.loss
>>> print(loss.item())
5.239729881286621
>>> print(f"{loss.item():.5f}")
5.23973
```"""
use_cache = use_cache if use_cache is not None else self.config.text_config.use_cache
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
......
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