Add a single line on how to use the trained checkpoints for sampling.
Add a single line on how to use the trained checkpoints for sampling.

PiperOrigin-RevId: 355824607

google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research