Searched refs:decode_one_token (Results 1 – 2 of 2) sorted by relevance
49 def decode_one_token( function57 return decode_one_token(model, x, **sampling_kwargs)[0]73 next_token, next_prob = decode_one_token(
87 def decode_one_token( function108 next_token, next_prob = decode_one_token(