One-line T5 fix left unmerged for 3 years: Hugging Face's closed PR on lm_head init

Birchlabs · x · 2026-10-21

Developer Birch-san notes Hugging Face never merged his one-line transformers PR (#26441) from 2023. It changes T5 lmhead weight init variance from 1 to the reciprocal of hiddendim: before, logits had variance hiddendim and initial cross-entropy loss of 110; after, variance 1 and initial loss 11, in line with expectations. Related to issue #16749; the PR remains closed and unmerged.

Original post →

More from Fun

Fun channel →