"""Implements a Hugging Face T5 wrapped inside a :class:`.ComposerModel`.""" from __future__ import annotations from typing import Mapping from transformers import AutoConfig, PreTrainedTokenizerBase, T5ForConditionalGeneration from .hf_fsdp import hf_get_init_device from .model_wrapper import HuggingFaceModelWithFSDP from .utils import init_empty_weights from .warnings import experimental_class