Unverified Commit 6df29ba5 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

fix wrong tokenizer checkpoint name in flax marian (#15391)


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 507601a5
...@@ -986,7 +986,7 @@ class FlaxMarianPreTrainedModel(FlaxPreTrainedModel): ...@@ -986,7 +986,7 @@ class FlaxMarianPreTrainedModel(FlaxPreTrainedModel):
```python ```python
>>> from transformers import MarianTokenizer, FlaxMarianMTModel >>> from transformers import MarianTokenizer, FlaxMarianMTModel
>>> tokenizer = MarianTokenizer.from_pretrained("facebook/marian-large-cnn") >>> tokenizer = MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-de")
>>> model = FlaxMarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-en-de") >>> model = FlaxMarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-en-de")
>>> text = "My friends are cool but they eat too many carbs." >>> text = "My friends are cool but they eat too many carbs."
...@@ -1054,7 +1054,7 @@ class FlaxMarianPreTrainedModel(FlaxPreTrainedModel): ...@@ -1054,7 +1054,7 @@ class FlaxMarianPreTrainedModel(FlaxPreTrainedModel):
>>> import jax.numpy as jnp >>> import jax.numpy as jnp
>>> from transformers import MarianTokenizer, FlaxMarianMTModel >>> from transformers import MarianTokenizer, FlaxMarianMTModel
>>> tokenizer = MarianTokenizer.from_pretrained("facebook/marian-large-cnn") >>> tokenizer = MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-de")
>>> model = FlaxMarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-en-de") >>> model = FlaxMarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-en-de")
>>> text = "My friends are cool but they eat too many carbs." >>> text = "My friends are cool but they eat too many carbs."
......
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