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

Merged

Ensure attention mask is on same device as inputs IDs during text generation #162

lewtun merged 3 commits into main from fix-generate

Add DistilGPT2 to unit tests

3e32d02e

Ensure attention mask on same device as input IDs

3df94f74

lewtun requested a review from

philschmid 3 years ago

lewtun requested a review from

michaelbenayoun 3 years ago

philschmid approved these changes on 2022-05-05

Add URL to source

751e8338

lewtun merged c9761991 into main 3 years ago

lewtun deleted the fix-generate branch 3 years ago

Reviewers

philschmid

michaelbenayoun

Assignees

No one assigned

Labels

None yet

Milestone

No milestone