"...gpu/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "ce3560a729e52f3a6f3a6120c248cd041e22093a"
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
Showing
Please register or sign in to comment