Kirill Gelvan commited on
Commit
a0f2756
1 Parent(s): 26ff0b0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -41,7 +41,8 @@ while True:
41
  input_user = input("===> Human: ")
42
 
43
  # encode the new user input, add parameters and return a tensor in Pytorch
44
- new_user_input_ids = tokenizer.encode(f"|0|{get_length_param(input_user, tokenizer)}|" + input_user + tokenizer.eos_token, return_tensors="pt")
 
45
  # append the new user input tokens to the chat history
46
  chat_history_ids = torch.cat([chat_history_ids, new_user_input_ids], dim=-1)
47
 
 
41
  input_user = input("===> Human: ")
42
 
43
  # encode the new user input, add parameters and return a tensor in Pytorch
44
+ new_user_input_ids = tokenizer.encode(f"|0|{get_length_param(input_user, tokenizer)}|" \
45
+ + input_user + tokenizer.eos_token, return_tensors="pt")
46
  # append the new user input tokens to the chat history
47
  chat_history_ids = torch.cat([chat_history_ids, new_user_input_ids], dim=-1)
48