changwangss's picture
update modeling_baichuan.py for torchscript mode with past_kv
d223af7