transformers
02a0b430 - Fixed TextGenerationPipeline on torch + GPU (#5629)

Commit
5 years ago
Fixed TextGenerationPipeline on torch + GPU (#5629) * Pytorch gpu => cpu proper device * Memoryless XLNet warning + fixed memories during generation * Revert "Memoryless XLNet warning + fixed memories during generation" This reverts commit 3d3251ff * Took the operations on the generated_sequence out of the ensure_device scope
Author
Parents
Loading