riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-16 19:18:24 +06:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Update Code example according to deprecation of AutoModeWithLMHead (#7555)

Browse Source 
'The class `AutoModelWithLMHead` is deprecated and will be removed in a future version. Please use `AutoModelForCausalLM` for causal language models, `AutoModelForMaskedLM` for masked language models and `AutoModelForSeq2SeqLM` for encoder-decoder models.'
I dont know how to change the 'How to use this model directly from the 🤗/transformers library:' part since it is not part of the model-paper
This commit is contained in:
Joshua H 2020-10-05 14:21:21 +02:00 committed by GitHub
parent 0d79de7322
commit 1a00f46c74
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model_cards/microsoft/DialoGPT-large/README.md
View File

@ -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! Now we are ready to try out how the model works as a chatting partner!
```python ```python
from transformers import AutoModelWithLMHead, AutoTokenizer from transformers import AutoModelForCausalLM, AutoTokenizer
import torch import torch
tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large") 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 # Let's chat for 5 lines
for step in range(5): for step in range(5):

4
model_cards/microsoft/DialoGPT-medium/README.md
View File

@ -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! Now we are ready to try out how the model works as a chatting partner!
```python ```python
from transformers import AutoModelWithLMHead, AutoTokenizer from transformers import AutoModelForCausalLM, AutoTokenizer
import torch import torch
tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium") 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 # Let's chat for 5 lines
for step in range(5): for step in range(5):

4
model_cards/microsoft/DialoGPT-small/README.md
View File

@ -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! Now we are ready to try out how the model works as a chatting partner!
```python ```python
from transformers import AutoModelWithLMHead, AutoTokenizer from transformers import AutoModelForCausalLM, AutoTokenizer
import torch import torch
tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-small") 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 # Let's chat for 5 lines
for step in range(5): for step in range(5):