r3dhummingbird
/

DialoGPT-small-neku

Text Generation Transformers PyTorch Safetensors gpt2 conversational Inference Endpoints text-generation-inference

Model card Files Files and versions Community

r3dhummingbird commited on Jun 7, 2021

Commit

4b8293d

•

1 Parent(s): b62b153

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -19,9 +19,9 @@ Chat with the model:
 ```python
 from transformers import AutoTokenizer, AutoModelWithLMHead
-tokenizer = AutoTokenizer.from_pretrained("r3dhummingbird/DialoGPT-medium-neku")
-model = AutoModelWithLMHead.from_pretrained("r3dhummingbird/DialoGPT-medium-neku")
 # Let's chat for 4 lines
 for step in range(4):

 ```python
 from transformers import AutoTokenizer, AutoModelWithLMHead
+tokenizer = AutoTokenizer.from_pretrained("r3dhummingbird/DialoGPT-small-neku")
+model = AutoModelWithLMHead.from_pretrained("r3dhummingbird/DialoGPT-small-neku")
 # Let's chat for 4 lines
 for step in range(4):