optimum
c9761991 - Ensure attention mask is on same device as inputs IDs during text generation (#162)

Comment changes are shownComment changes are hidden
Commit
3 years ago
Ensure attention mask is on same device as inputs IDs during text generation (#162) * Add DistilGPT2 to unit tests * Ensure attention mask on same device as input IDs * Add URL to source
Author
Parents
  • optimum/onnxruntime
    • File
      modeling_ort.py
  • tests/onnxruntime
    • File
      test_modeling_ort.py