Fix comment

This commit is contained in:
Matt Clayton 2025-02-10 12:32:00 -05:00
parent 4df23e961c
commit 9e28bedc81

View File

@ -76,7 +76,7 @@ class GenerationResponse:
Args:
text (str): The next segment of decoded text. This can be an empty string.
token (int): The next token.
from_draft (bool): Whether the token was generated by a draft model.
from_draft (bool): Whether the token was generated by the draft model.
logprobs (mx.array): A vector of log probabilities.
prompt_tokens (int): The number of tokens in the prompt.
prompt_tps (float): The prompt processing tokens-per-second.