Update README.md

README.md

@@ -89,7 +89,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]:]