Method __init__

demo/HuggingFace/T5/export.py:227–231

(self, encoder)

    class TorchModule(Module, GenerationMixin):
        def __init__(self, encoder):
            super().__init__()
            self.encoder = encoder
            # Use hardcoded value to extend compatibility with older HF versions.
            self.main_input_name = "input_ids"

        def forward(self, *input, **kwargs):
            return self.encoder(*input, **kwargs)[0]