optimum-graphcore
optimum-graphcore copied to clipboard
Implement the different generation methods using poptorch.for_loop to make things much faster
- [ ] Greedy search
- [ ] Sample
- [ ] Beam search
- [ ] Beam sample
- [ ] Group beam search
- [ ] Constrained beam search