thunlp / OpenPrompt

An Open-Source Framework for Prompt-Learning.
https://thunlp.github.io/OpenPrompt/
Apache License 2.0
4.36k stars 454 forks source link

Some question in ./prompts/prefix_tuning_template.py #209

Open BAOOOOOM opened 2 years ago

BAOOOOOM commented 2 years ago

In process_batch module, why creat a past_key_values in batch_size=1 and then use expand to batchsize?

BAOOOOOM commented 2 years ago

image