transformers
c38f4e1f - Running a pipeline of `float16`. (#17637)

Commit
3 years ago
Running a pipeline of `float16`. (#17637) When we're preparing the tensors for CPU for postprocessing, we need to upgrade the `float16` to `float32` since CPUs don't have instructions for `[b]float16`.
Author
Parents
Loading