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