Skip to content

Commit b36ea66

Browse files
committed
Fixing typo causing error.
1 parent d904304 commit b36ea66

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/forge/actors/reference_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ async def forward(
128128
Args:
129129
input_ids (torch.Tensor): input token ids with shape [group_size, req + res length].
130130
max_req_tokens (int): maximum request length.
131-
return_logprobs (bool): whether to return og probabilities instead of raw logits.
131+
return_logprobs (bool): whether to return log probabilities instead of raw logits.
132132
133133
return_logprobs flag significantly impacts the amount of data transferred to the caller:
134134
- When False: Returns logits with shape [group_size, req + res_length, vocab_size].

0 commit comments

Comments
 (0)