SimCTG icon indicating copy to clipboard operation
SimCTG copied to clipboard

Modify to support batch size N

Open quinle opened this issue 2 years ago • 2 comments

Hi, Thank you so much for the great project, it is amazing. This PR is for adding support for OPT inference batch size N. I'm looking forward to your review ^^

quinle avatar Oct 14 '22 09:10 quinle

Hi @quinle -- Thank you for your interest in our work and the great effort! We will take a closer look to your commits, many thanks!

@gmftbyGMFTBY Are you able to review the PR by any chance?

yxuansu avatar Oct 19 '22 10:10 yxuansu

Hi @quinle -- Thank you again for your interest in our work. We are glad that Huggingface has officially supported contrastive search in transformers. Here, we provide a short tutorial. In transformers, contrastive search supports batch inference by default.

Hope this resource could help you :-)

Best,

Yixuan

yxuansu avatar Nov 05 '22 10:11 yxuansu