Remove faulty call with device to t5 forward method

joecummings · joecummings · commit 5cb9e10381bd · 2022-10-14T14:52:02.000-04:00
diff --git a/torchtext/prototype/models/t5/modules.py b/torchtext/prototype/models/t5/modules.py
@@ -258,7 +258,6 @@ def _t5_multi_head_attention_forward(
                     tgt_len,
                     src_len,
                     bidirectional=(not self.is_decoder),
-                    device=k.device,
                 )
 
         # Calculate attention and out projection

Original file line number	Diff line number	Diff line change
`@@ -258,7 +258,6 @@ def _t5_multi_head_attention_forward(`
`258`	`258`	`tgt_len,`
`259`	`259`	`src_len,`
`260`	`260`	`bidirectional=(not self.is_decoder),`
`261`		`- device=k.device,`
`262`	`261`	`)`
`263`	`262`
`264`	`263`	`# Calculate attention and out projection`