Method encode

t5/t5.py:319–320

(self, inputs: mx.array)

        self.model_dim = config.d_model

    def encode(self, inputs: mx.array):
        return self.encoder(self.wte(inputs))

    def decode(
        self,