Minor correction to demo transformer script#21033

Borda merged 6 commits intoLightning-AI:masterLightning-AI/pytorch-lightning:masterfrom

SkafteNicki:demo_transformer_fixSkafteNicki/pytorch-lightning:demo_transformer_fixCopy head branch name to clipboard