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

Merged

Commits

Add DistilGPT2 to unit tests

lewtun committed 3 years ago
Ensure attention mask on same device as input IDs

lewtun committed 3 years ago
Add URL to source

lewtun committed 3 years ago

Loading