Skip to content

Commit

Permalink
fix typo in README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Fan Zhang committed Oct 24, 2024
1 parent 2851090 commit f6ea9aa
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -213,7 +213,7 @@ GENERATION_CONFIG = GenerationConfig(
outputs = model.generate(
inputs.input_ids.to("cuda:0"),
GENERATION_CONFIG,
attention_mask=pos_inputs.attention_mask.to("cuda:0"),
attention_mask=inputs.attention_mask.to("cuda:0"),
)

outputs = outputs[:, inputs.input_ids.shape[-1]:]
Expand Down

0 comments on commit f6ea9aa

Please sign in to comment.