byeongal commited on
Commit
228f755
·
1 Parent(s): b5cf795

update model and README.md

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. 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(5):
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:42e11fe9d1a801b2d192e27950a43f8e6b12a5f7a389d682a64bc2804087a8f5
3
  size 513305211
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6dec0cdd37c1d0af26f80bacf0c692e2fdd8045be4617042920d2ffa6104b4df
3
  size 513305211