Fix a typo

yoshitomo-matsubara · yoshitomo-matsubara · commit 12ad9c9ae0ea · 2025-02-23T20:51:08.000-08:00
diff --git a/examples/hf_transformers/custom/forward_proc.py b/examples/hf_transformers/custom/forward_proc.py
@@ -10,7 +10,7 @@ def forward_batch_as_kwargs(model, sample_batch, targets=None, supp_dict=None):
 def forward_batch_as_kwargs_for_distillbert(model, sample_batch, targets=None, supp_dict=None):
     inputs = {
         'input_ids': sample_batch[0],
-        'attentio_mask': sample_batch[1],
+        'attention_mask': sample_batch[1],
         'start_positions': sample_batch[3],
         'end_positions': sample_batch[4],
     }

Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,7 @@ def forward_batch_as_kwargs(model, sample_batch, targets=None, supp_dict=None):`
`10`	`10`	`def forward_batch_as_kwargs_for_distillbert(model, sample_batch, targets=None, supp_dict=None):`
`11`	`11`	`inputs = {`
`12`	`12`	`'input_ids': sample_batch[0],`
`13`		`- 'attentio_mask': sample_batch[1],`
	`13`	`+ 'attention_mask': sample_batch[1],`
`14`	`14`	`'start_positions': sample_batch[3],`
`15`	`15`	`'end_positions': sample_batch[4],`
`16`	`16`	`}`