Update README.md

README.md

@@ -31,32 +31,41 @@ tags:
 >>>     print("1",prompt,"2")
 >>>     input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(device)
 >>>     generate_ids = model.generate(input_ids=input_ids,
->>>     max_length=4096,
+>>>     max_length=2048,
+>>>     #  do_sample = True, 
+>>>     # eos_token_id=tokenizer.eos_token_id )
 >>>     num_beams=1,
->>>     do_sample=True, top_p=0.9, temperature=0.95, repetition_penalty=1.05, eos_token_id=tokenizer.eos_token_id)
->>>     output = tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
->>>     response = output[len(prompt):]
+>>>     do_sample=True, top_p=0.9, temperature=0.95, repetition_penalty=1.05,eos_token_id=tokenizer.eod_id, 
+>>>             bos_token_id=tokenizer.im_start_id, 
+>>>             pad_token_id=tokenizer.eod_id)
+>>>     output = tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]#
+>>>     response = output.split("牧濑红莉栖:")[-1]
+>>>     print(output)
+>>>    # print("1",response,"2")
 >>>     return response
 >>> device = torch.device('cuda')
+>>> print(tokenizer.encode("</s>"))
 >>> history=[]
 >>> max_history_len=12
->>> rating="0"
 >>> while True:
 >>>     text=input("user:")
->>>     text=f"人类:{text}</s>"
+>>>     text=f"人类:{text}<|endoftext|>"
 >>>     history.append(text)
->>>     input_text="爱丽丝的人格:你叫爱丽丝，是一个傲娇，腹黑的16岁少女<|im_end|>"
+>>>     input_text=""
+>>>     input_text="牧濑红莉栖的人格:你是牧濑红莉栖来自《命运石之门》<|endoftext|>\n"
 >>>     for history_id, history_utr in enumerate(history[-max_history_len:]):
 >>>         input_text = input_text + history_utr + '\n'
->>>     prompt = input_text+"爱丽丝:"
+>>>     prompt = input_text+"牧濑红莉栖:"
 >>>     prompt =prompt.strip()
 >>>     response = generate(prompt)
 >>>     response=response.strip()
->>>     response="爱丽丝:"+response+"<|im_end|>"
+>>>     response="牧濑红莉栖:"+response+"<|endoftext|>"
 >>>     print("1",response,"2")
 >>>     history.append(response)
-人类:我还要去上班
-爱丽丝:哎呀呀~这么无聊，竟然还要去工作？
+
+牧濑红莉栖的人格:你是牧濑红莉栖来自《命运石之门》<|endoftext|>
+人类:你是谁<|endoftext|>
+牧濑红莉栖:我叫牧濑红莉栖，是一名来自《命运石之门》的科学家。我的工作是研究时间旅行技术，并试图挽救世界。然而，由于一些未知的原因，我现在被困在一个名为“Labo”的实验中，无法离开。我需要你的帮助，才能找到回到现实世界的方法。<|endoftext|>
 ```
 ## 引用
 ```