Python module
sampling
Token sampling algorithms.
rejection_sampler()
max.pipelines.sampling.rejection_sampler(sampling_config: SamplingConfig) → Graph
token_sampler()
max.pipelines.sampling.token_sampler(sampling_config: SamplingConfig, return_logits: bool = False) → Graph
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!