update model and README.md
Browse files- README.md +1 -1
- pytorch_model.bin +1 -1
README.md
CHANGED
@@ -16,7 +16,7 @@ import torch
|
|
16 |
tokenizer = AutoTokenizer.from_pretrained("byeongal/Ko-DialoGPT")
|
17 |
model = AutoModelForCausalLM.from_pretrained("byeongal/Ko-DialoGPT")
|
18 |
|
19 |
-
for step in range(
|
20 |
# encode the new user input, add the eos_token and return a tensor in Pytorch
|
21 |
new_user_input_ids = tokenizer.encode(input(">> User:") + tokenizer.eos_token, return_tensors='pt')
|
22 |
# append the new user input tokens to the chat history
|
|
|
16 |
tokenizer = AutoTokenizer.from_pretrained("byeongal/Ko-DialoGPT")
|
17 |
model = AutoModelForCausalLM.from_pretrained("byeongal/Ko-DialoGPT")
|
18 |
|
19 |
+
for step in range(2):
|
20 |
# encode the new user input, add the eos_token and return a tensor in Pytorch
|
21 |
new_user_input_ids = tokenizer.encode(input(">> User:") + tokenizer.eos_token, return_tensors='pt')
|
22 |
# append the new user input tokens to the chat history
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 513305211
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6dec0cdd37c1d0af26f80bacf0c692e2fdd8045be4617042920d2ffa6104b4df
|
3 |
size 513305211
|