Aero commited on
Commit
4addf3e
1 Parent(s): 30b0b35

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -8,8 +8,8 @@ license: mit
8
  ```python
9
  from transformers import AutoTokenizer, AutoModelWithLMHead
10
 
11
- tokenizer = AutoTokenizer.from_pretrained("Aero/Tsubomi-Haruno")
12
- model = AutoModelWithLMHead.from_pretrained("Aero/Tsubomi-Haruno")
13
  # Let's chat for 4 lines
14
  for step in range(4):
15
  # encode the new user input, add the eos_token and return a tensor in Pytorch
@@ -30,4 +30,4 @@ for step in range(4):
30
 
31
  # pretty print last ouput tokens from bot
32
  print("Tsubomi: {}".format(tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)))
33
- ```
8
  ```python
9
  from transformers import AutoTokenizer, AutoModelWithLMHead
10
 
11
+ tokenizer = AutoTokenizer.from_pretrained("r3dhummingbird/DialoGPT-medium-joshua")
12
+ model = AutoModelWithLMHead.from_pretrained("r3dhummingbird/DialoGPT-medium-joshua")
13
  # Let's chat for 4 lines
14
  for step in range(4):
15
  # encode the new user input, add the eos_token and return a tensor in Pytorch
30
 
31
  # pretty print last ouput tokens from bot
32
  print("Tsubomi: {}".format(tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)))
33
+ ```