diff --git a/model_cards/microsoft/DialoGPT-large/README.md b/model_cards/microsoft/DialoGPT-large/README.md
index 875d8417003..0c69cc60a85 100644
--- a/model_cards/microsoft/DialoGPT-large/README.md
+++ b/model_cards/microsoft/DialoGPT-large/README.md
@@ -31,12 +31,12 @@ ArXiv paper: [https://arxiv.org/abs/1911.00536](https://arxiv.org/abs/1911.00536
 Now we are ready to try out how the model works as a chatting partner!
 
 ```python
-from transformers import AutoModelWithLMHead, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
-model = AutoModelWithLMHead.from_pretrained("microsoft/DialoGPT-large")
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
 
 # Let's chat for 5 lines
 for step in range(5):
diff --git a/model_cards/microsoft/DialoGPT-medium/README.md b/model_cards/microsoft/DialoGPT-medium/README.md
index 9fc35ce20ae..eb41fb7b0e8 100644
--- a/model_cards/microsoft/DialoGPT-medium/README.md
+++ b/model_cards/microsoft/DialoGPT-medium/README.md
@@ -31,12 +31,12 @@ ArXiv paper: [https://arxiv.org/abs/1911.00536](https://arxiv.org/abs/1911.00536
 Now we are ready to try out how the model works as a chatting partner!
 
 ```python
-from transformers import AutoModelWithLMHead, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
-model = AutoModelWithLMHead.from_pretrained("microsoft/DialoGPT-medium")
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 
 # Let's chat for 5 lines
 for step in range(5):
diff --git a/model_cards/microsoft/DialoGPT-small/README.md b/model_cards/microsoft/DialoGPT-small/README.md
index d2fd0aea0b8..811ff26fb0c 100644
--- a/model_cards/microsoft/DialoGPT-small/README.md
+++ b/model_cards/microsoft/DialoGPT-small/README.md
@@ -31,12 +31,12 @@ ArXiv paper: [https://arxiv.org/abs/1911.00536](https://arxiv.org/abs/1911.00536
 Now we are ready to try out how the model works as a chatting partner!
 
 ```python
-from transformers import AutoModelWithLMHead, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-small")
-model = AutoModelWithLMHead.from_pretrained("microsoft/DialoGPT-small")
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-small")
 
 # Let's chat for 5 lines
 for step in range(5):