demonstration generation

Dear developers,

Thanks for the work!

I would like to try auto-cot in a custom task and a dataset. I couldn't get which code refers the automatic demonstration generation. Could you please help me on this?

Thanks in advance.