update modeling_seq2seq.py

huggingface · Aug 21, 2023 · 7d27fdb · 7d27fdb
1 parent b500f22
commit 7d27fdb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/onnxruntime/modeling_seq2seq.py b/optimum/onnxruntime/modeling_seq2seq.py
@@ -1486,8 +1486,8 @@ def forward(
                 flattened_patches=flattened_patches,
                 attention_mask=attention_mask,
             )
-        # work around for Unexpected input data type (tensor(float)) for attention_mask
-        # Need to be looked into
+
+        #decoder requires torch.LongTensor for attention_mask
         attention_mask = attention_mask.to(torch.int64)
         # Decode