optimum
c9761991
- Ensure attention mask is on same device as inputs IDs during text generation (#162)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Hide Minimap (CTRL+M)
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
References
#162 - Ensure attention mask is on same device as inputs IDs during text generation
Author
lewtun
Parents
1c4b5b1c
Files
2
optimum/onnxruntime
modeling_ort.py
tests/onnxruntime
test_modeling_ort.py
Loading