transformers

01cbbf31 - Make logits match original implementation