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