optimum

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

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