cerebras.modelzoo.common.utils.model.generation_utils#

Functions

sample_tokens

Function used to sample tokens, if needed.