Skip to content

Commit

Permalink
Apply isort and black reformatting
Browse files Browse the repository at this point in the history
Signed-off-by: shengyangs <[email protected]>
  • Loading branch information
shengyangs committed Jul 17, 2024
1 parent 4cd11c4 commit fd3d7b8
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ def get_prompt_template_example(special_tokens):


def identify_start_index_of_subsequence(subsequence, sequence):
""" find the location of the small tensor in the large tensor.
"""find the location of the small tensor in the large tensor.
e.g. small = [1,3], large = [2,3,1,3], returns 2
small = [3,2], large = [2,3,1,3], returns -1
Args:
Expand All @@ -100,7 +100,7 @@ def _mask_targets(
label_start_ids,
num_turn_start_tokens,
):
""" This function masks the tokens so the loss is computed only on the non-masked role's responses.
"""This function masks the tokens so the loss is computed only on the non-masked role's responses.
For 'TEXT_TO_VALUE' type, the loss is computed on the value attributes.
Args:
Expand Down

0 comments on commit fd3d7b8

Please sign in to comment.