MEDT commited on
Commit
97bbb40
1 Parent(s): 996a511

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -25,8 +25,8 @@ tokenizer = AutoTokenizer.from_pretrained("r3dhummingbird/DialoGPT-medium-joshua
25
 
26
  model = AutoModelWithLMHead.from_pretrained("r3dhummingbird/DialoGPT-medium-joshua")
27
 
28
- # Let's chat for 4 lines
29
- for step in range(4):
30
  # encode the new user input, add the eos_token and return a tensor in Pytorch
31
  new_user_input_ids = tokenizer.encode(input(">> User:") + tokenizer.eos_token, return_tensors='pt')
32
  # print(new_user_input_ids)
 
25
 
26
  model = AutoModelWithLMHead.from_pretrained("r3dhummingbird/DialoGPT-medium-joshua")
27
 
28
+ # Let's chat for 100 lines
29
+ for step in range(100):
30
  # encode the new user input, add the eos_token and return a tensor in Pytorch
31
  new_user_input_ids = tokenizer.encode(input(">> User:") + tokenizer.eos_token, return_tensors='pt')
32
  # print(new_user_input_ids)