+ UV-j^RIHtRRltR#)NcpV^8dQhR\PR\R\R\P/#)logitstop_p temperaturereturn)mxarrayfloat)formats"e/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_vlm/sample_utils.py __annotate__rs0+4+4288+4E+4+4"((+4cVP^8HpV'd VR,pVP\P8Xd VP \P 4p\P !W, RR7p\P!VRR7p\P!WERR7p\P!VRR7p\P!V^V, 8V\P!V44p\PP\P!V44p \P!WYR,RR7PR4p V'dV P^4#T #)a Apply top-p (nucleus) sampling to logits. Args: logits: The logits from the model's output. Shape [..., vocab]; commonly [vocab], [B, vocab], or [B, T, vocab] (e.g. MTP speculative verify output). top_p: The cumulative probability threshold for top-p filtering. temperature: Temperature parameter for softmax distribution reshaping. Returns: token selected based on the top-p criterion. Shape matches logits with the trailing vocab axis removed (e.g. [], [B], or [B, T]). N)axis).N)ndimdtyper bfloat16astypefloat32softmaxargsorttake_along_axiscumsumwhere zeros_likerandom categoricallogsqueeze) rrr unbatchedprobssorted_indices sorted_probscumulative_probs top_probs sampled_postokens &&& r top_p_samplingr*s q I   #rzz* JJv+" 5EZZB/N%%e"ELyyB71u9$ l#I ))''y(9:K   ~9/EB O W W  E )5== 3e3r)mlx.corecorer r*rr r.s +4r