How to do temperature rescaling

[Lifedecoder]

Hi!
Thanks for this marvellous work.
I have questions about how to do temperature scaling, like Fig. 6 in the paper does. I use pretrained ODE model to generate 256x256 Celeba images, but the results are too smooth and blurry. I wonder how I can get more realistic pics like Fig. 6 does? Specifically how to "reduce the norm of embedding" to do temperature scaling?

Many thanks.