Fix FLAX_MULTIPLE_CHOICE_SAMPLE typo (#14871)

eec9c8bb · Mishig Davaadorj · GitHub · e51c7b58 · eec9c8bb
Unverified Commit eec9c8bb authored Dec 21, 2021 by Mishig Davaadorj Committed by GitHub Dec 21, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/transformers/file_utils.py src/transformers/file_utils.py +2 -1

No files found.
--- a/src/transformers/file_utils.py
+++ b/src/transformers/file_utils.py
@@ -1491,7 +1491,8 @@ FLAX_MULTIPLE_CHOICE_SAMPLE = r"""
    >>> encoding = tokenizer([prompt, prompt], [choice0, choice1], return_tensors='jax', padding=True)
    >>> outputs = model(**{{k: v[None, :] for k,v in encoding.items()}})

-    >>> logits = outputs.logits ```
+    >>> logits = outputs.logits
+    ```
 """

 FLAX_CAUSAL_LM_SAMPLE = r"""