cCiSrSSKJr SSKJr SSKJrJrJ r SSK J r \R"\ 5rS rS \R S \S \S \R 4Sjr"SS\ 5r"SS\5r"SS\5r/SQrg)zFlax mT5 model.N)logging)FlaxT5EncoderModelFlaxT5ForConditionalGeneration FlaxT5Model) MT5ConfigT5Config input_ids pad_token_iddecoder_start_token_idreturnc[R"U5nURSS2SS24RUSS2SS245nURSS2S4RU5n[R"US:HX5nU$)z) Shift input ids one token to the right. Nr ri)jnp zeros_likeatsetwhere)r r rshifted_input_idss c/home/james-whalen/.local/lib/python3.13/site-packages/transformers/models/mt5/modeling_flax_mt5.pyshift_tokens_rightrsy1),,QU377 !SbS&8IJ),,QT2667MN "3t";\] c \rSrSrSrSr\rSrg) FlaxMT5Model*a This class overrides [`FlaxT5Model`]. Please check the superclass for the appropriate documentation alongside usage examples. Examples: ```python >>> from transformers import FlaxMT5Model, AutoTokenizer >>> model = FlaxMT5Model.from_pretrained("google/mt5-small") >>> tokenizer = AutoTokenizer.from_pretrained("google/mt5-small") >>> article = "UN Offizier sagt, dass weiter verhandelt werden muss in Syrien." >>> summary = "Weiter Verhandlung in Syrien." >>> inputs = tokenizer(article, return_tensors="np") >>> decoder_input_ids = tokenizer(text_target=summary, return_tensors="np").input_ids >>> outputs = model(input_ids=inputs["input_ids"], decoder_input_ids=decoder_input_ids) >>> hidden_states = outputs.last_hidden_state ```mt5N __name__ __module__ __qualname____firstlineno____doc__ model_typer config_class__static_attributes__rrrrr* ,JLrrc \rSrSrSrSr\rSrg)FlaxMT5EncoderModelEa This class overrides [`FlaxT5EncoderModel`]. Please check the superclass for the appropriate documentation alongside usage examples. Examples: ```python >>> from transformers import FlaxT5EncoderModel, AutoTokenizer >>> model = FlaxT5EncoderModel.from_pretrained("google/mt5-small") >>> tokenizer = AutoTokenizer.from_pretrained("google/mt5-small") >>> article = "UN Offizier sagt, dass weiter verhandelt werden muss in Syrien." >>> summary = "Weiter Verhandlung in Syrien." >>> inputs = tokenizer(article, return_tensors="np") >>> decoder_input_ids = tokenizer(text_target=summary, return_tensors="np").input_ids >>> outputs = model(input_ids=inputs["input_ids"]) >>> hidden_states = outputs.last_hidden_state ```rrNr rrrr+r+Er)rr+c \rSrSrSrSr\rSrg)FlaxMT5ForConditionalGeneration`a This class overrides [`FlaxT5ForConditionalGeneration`]. Please check the superclass for the appropriate documentation alongside usage examples. Examples: ```python >>> from transformers import FlaxMT5ForConditionalGeneration, AutoTokenizer >>> model = FlaxMT5ForConditionalGeneration.from_pretrained("google/mt5-small") >>> tokenizer = AutoTokenizer.from_pretrained("google/mt5-small") >>> article = "UN Offizier sagt, dass weiter verhandelt werden muss in Syrien." >>> summary = "Weiter Verhandlung in Syrien." >>> inputs = tokenizer(article, return_tensors="np") >>> decoder_input_ids = tokenizer(text_target=summary, return_tensors="np").input_ids >>> outputs = model(**inputs, decoder_input_ids=decoder_input_ids) >>> logits = outputs.logits ```rrNr rrrr.r.`r)rr.)r+r.r)r% jax.numpynumpyrutilsrt5.modeling_flax_t5rrrconfiguration_mt5r get_loggerr!logger_CONFIG_FOR_DOCndarrayintrrr+r.__all__rrrr;saa(   H % #++ S Z] bebmbm ;6,6&D6 Ur