Function get_chatrounds_ids

src/models/test.py:69–74

(chatrounds_list)

def get_chatrounds_ids(chatrounds_list):
    input_ids  = []
    for chatround in chatrounds_list:
        input_ids += tokenizer.encode(role_start + chatround["role"]+ role_end) + tokenizer.encode(chatround["content"], add_special_tokens=False) + [tokenizer.eos_token_id]
    input_ids += tokenizer.encode(role_start + "bot" + role_end)
    return input_ids

class GetAssistantAns():
    # 按照自己推理需求自己修改代码