Skip to content

Commit

Permalink
Update composer/datasets/in_context_learning_evaluation.py
Browse files Browse the repository at this point in the history
fix spelling

Co-authored-by: Daniel King <[email protected]>
  • Loading branch information
maxisawesome and dakinggg authored Jan 25, 2024
1 parent 5ff9a30 commit c44c763
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion composer/datasets/in_context_learning_evaluation.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ def _trim_context(context_enc: List, continuation_enc: List, max_seq_len: int) -

def _get_continuation_span(context_enc: List, continuation_enc: List) -> torch.Tensor:
"""
Gets the list of indices of the continutaion tokens for language modeling or generaiton tasks.
Gets the list of indices of the continuation tokens for language modeling or generation tasks.
Args:
context_enc (list): list of context tokens
Expand Down

0 comments on commit c44c763

Please sign in to comment.