Update README.md

README.md

@@ -55,27 +55,18 @@ for step in range(100):
     chat_history.append("A: " + user_input)
     while len(chat_history) > 5:
         chat_history.pop(0)
-    # print(chat_history)
     hist = ""
     for chat in chat_history:
         hist += "\n" + chat
     hist += "\nB: "
-    # print("====", len(chat_history))
-    # print("===>", hist.replace("\n", " / "))
-    # print("----")
-    # encode the new user input, add the eos_token and return a tensor in Pytorch
     new_user_input_ids = tokenizer.encode(hist, return_tensors='pt')
-    # print(new_user_input_ids)
 
-    # append the new user input tokens to the chat history
-    #bot_input_ids = torch.cat([chat_history_ids, new_user_input_ids], dim=-1) if step > 0 else new_user_input_ids
     bot_input_ids = new_user_input_ids
 
     # generated a response while limiting the total chat history to 1000 tokens, 
     chat_history_ids = model.generate(
         bot_input_ids, max_length=200,
         pad_token_id=tokenizer.eos_token_id,  
-        #no_repeat_ngram_size=3,       
         do_sample=True, 
         #top_k=100, 
         #top_p=0.7,