Views
No views yet
loss_likelihood与loss_contrast的比例基本保持在9:1至8:2之间,根据原论文的实验,这可能并未是最佳配比。ROUGE_1、ROUGE_2、ROUGE_L的平均值进行排序。1from transformers import AutoModelForCausalLM, AutoTokenizer
2model_name = "jiangchengchengNLP/qwen_0.5B_instruct_law_summarize"
3model = AutoModelForCausalLM.from_pretrained(
4 model_name,
5 torch_dtype="auto",
6 device_map="auto",
7)
8tokenizer = AutoTokenizer.from_pretrained(model_name)
9with open("河北宁晋农村商业银行股份有限公司小刘村支行与杨保会、杨立田等借款合同纠纷一审民事判决书.txt", "r", encoding="utf-8") as f:
10 prompt = f.read()
11messages = [
12 {"role": "system", "content": "你是法律小助手,你的任务是将用户输入的内容整理成一份摘要"},
13 {"role": "user", "content": prompt}
14]
15text = tokenizer.apply_chat_template(
16 messages,
17 tokenize=False,
18 add_generation_prompt=True
19)
20model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
21generated_ids =model.generate(
22 **model_inputs,
23 max_new_tokens=512
24)
25generated_ids = [
26 output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
27]
28response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
29print(response)原被告系借款合同关系。原告提出诉求:1、请求法院依法判令被告偿还借款本金2000元及利息4912.6元;2、本案诉讼费用由被告承担。经审理查明,双方签订的借款合同和保证担保借款合同合法有效,被告未按约定偿还原告借款本息,应负违约责任。依据《中华人民共和国合同法》第八条、第六十条、第二百零五条、第二百零六条、二百零七条,《中华人民共和国担保法》第十八条、第二十一条以及《中华人民共和国民事诉讼法》第一百四十四条之规定,判决:1、被告偿还借款本息24912.6元;2、被告承担连带清偿责任。ROUGE_1:0.5703008654122925ROUGE_2:0.4040075748793321ROUGE_L:0.47319922532178305