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