cerebras.modelzoo.common.utils.model.generation_utils

Functions

sample_tokens

Function used to sample tokens, if needed.

previous

cerebras.modelzoo.common.utils.model.attention_utils.StridedSparseAttentionBuilder

next

cerebras.modelzoo.common.utils.model.generation_utils.sample_tokens