Pre-silence Pyre Errors for upcoming upgrade] [batch:67/603] [shard:3/N]

Reviewed By: MaggieMoss

Differential Revision: D65290095

fbshipit-source-id: ced87d096aa8939700de5599ce6984cd7ae93912
This commit is contained in:
generatedunixname89002005287564 2024-10-31 16:26:25 -07:00 committed by Facebook GitHub Bot
parent 9eaed4c495
commit 008c7ab58c

View File

@ -209,6 +209,7 @@ def handle_seq_id(
seq_id = torch.tensor(seq_id, dtype=torch.long, device=device)
# pyre-fixme[16]: Item `List` of `Union[List[int], List[str], LongTensor]` has
# no attribute `to`.
# pyre-fixme[7]: Expected `LongTensor` but got `Tensor`.
return seq_id.to(device)