Hoagy

Do you mean in the model that I perform the layer norm on, so I don't need to change any of this code?

Exactly

Looks like its being caused by something in lsp#internal#completion#documentation#_enable since I can prevent it by setting g:lsp_completion_documentation_enabled = 0