Skip to content

Commit

Permalink
Merge pull request #19 from eltociear/patch-1
Browse files Browse the repository at this point in the history
Fix typo in constrained_generator.py
  • Loading branch information
renqianluo authored Feb 3, 2023
2 parents aba073d + e1f310a commit 4cf20c3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/constrained_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -282,7 +282,7 @@ def _generate(
finalized = torch.jit.annotate(
List[List[Dict[str, Tensor]]],
[torch.jit.annotate(List[Dict[str, Tensor]], []) for i in range(bsz)],
) # contains lists of dictionaries of infomation about the hypothesis being finalized at each step
) # contains lists of dictionaries of information about the hypothesis being finalized at each step

# a boolean array indicating if the sentence at the index is finished or not
finished = [False for i in range(bsz)]
Expand Down

0 comments on commit 4cf20c3

Please sign in to comment.