DeepSpeed
0fc3daad
- Add position_ids arg to OPTEmbedding forward function (#6939)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
350 days ago
Add position_ids arg to OPTEmbedding forward function (#6939) This PR updates the DeepSpeed `OPTEmbedding` forward function to include a new `positions_ids` argument. --------- Co-authored-by: Logan Adams <loadams@microsoft.com>
References
#6939 - Add position_ids arg to OPTEmbedding forward function
Author
lekurile
Parents
45fce45c
Loading