Kirill Gelvan
commited on
Commit
•
a0f2756
1
Parent(s):
26ff0b0
Update README.md
Browse files
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)}|"
|
|
|
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 |
|