Method __init__

ioblocks.py:87–89

(self, dim, vocab_size)

class GPTOutput(nn.Module):
    def __init__(self, dim, vocab_size):
        super().__init__()
        self.output = nn.Linear(dim, vocab_size, bias=False)

    def forward(self, x):
        return self.output(x)