--- library_name: keras-hub license: apache-2.0 tags: - text-classification - keras pipeline_tag: text-generation --- ### Model Overview ⚠️ T5 is currently only available via the `keras-hub-nightly` package. Use `pip install keras-hub-nightly` to try this model. T5 encoder-decoder backbone model. T5 is a LLM pretrained on a mix of unsupervised and supervised tasks, where each task is converted to a sequence-to-sequence format. T5 works well on a variety of tasks out-of-the-box by prepending various prefixex to the input sequence, e.g., for translation: `"translate English to German: ..."`, for summarization: `"summarize: ..."`. T5 was introduced in [Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer](https://arxiv.org/abs/1910.10683) The default constructor gives a fully customizable, randomly initialized T5 model with any number of layers, heads, and embedding dimensions. To load preset architectures and weights, use the `from_preset` constructor. Disclaimer: Pre-trained models are provided on an "as is" basis, without warranties or conditions of any kind. __Arguments__ - __vocabulary_size__: int. The size of the token vocabulary. - __num_layers__: int. The number of Transformer layers. - __num_heads__: int. The number of attention heads for each Transformer. The hidden size must be divisible by the number of attention heads. - __hidden_dim__: int. The hidden size of the Transformer layers. - __intermediate_dim__: int. The output dimension of the first Dense layer in a two-layer feedforward network for each Transformer layer. - __key_value_dim__: int. The dimension of each head of the key/value projections in the multi-head attention layers. Defaults to hidden_dim / num_heads. - __dropout__: float. Dropout probability for the Transformer layers. - __activation__: activation function (or activation string name). The activation to be used in the inner dense blocks of the Transformer layers. Defaults to `"relu"`. - __use_gated_activation__: boolean. Whether to use activation gating in the inner dense blocks of the Transformer layers. The original T5 architecture didn't use gating, but more recent versions do. Defaults to `True`. - __layer_norm_epsilon__: float. Epsilon factor to be used in the layer normalization layers in the Transformer layers. - __tie_embedding_weights__: boolean. If `True`, the weights of the token embedding and the weights projecting language model outputs from `hidden_dim`