Method __init__

demo/HuggingFace/T5/export.py:195–198

(self, decoder, device = "cpu")

class T5DecoderCrossAttentionKVGenerator(Module):
    def __init__(self, decoder, device = "cpu"):
        super().__init__()
        self.decoder = decoder
        self.device = device

    def forward(self, encoder_hidden_states):
        '''