↓ 2 callersFunctionrollout(
client: openai.AsyncOpenAI,
model: art.TrainableModel,
prompt: str,
*,
max_tokens: int,
dev/yes-no-maybe-metrics.py:94
↓ 1 callersFunction_calculate_logprobs(
lm_head_t: torch.Tensor, # Shape [H, V]
hidden_states: torch.Tensor, # Shape [B, S, H]
next_in
src/art/unsloth/train.py:356