transformers
99d67f1a - Improve greedy search memory usage (#32895)

Commit

1 year ago

Improve greedy search memory usage (#32895) Do not call torch.repeat_interleave if expand_size is 1

References

Author

regisss

Parents