The target's draft projection: embed and the head table (logits). Borrowed from the model.
embed
logits
The target's draft projection:
embedand the head table (logits). Borrowed from the model.