gsarti commited on
Commit
d97d1e5
1 Parent(s): f979ca2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -48,10 +48,10 @@ For a list of individual model parameters, refer to the `config.json` file in th
48
  ## Using the models
49
 
50
  ```python
51
- from transformers import T5Tokenizer, T5ForConditionalGeneration
52
 
53
- tokenizer = T5Tokenizer.from_pretrained("gsarti/it5-large")
54
- model = T5ForConditionalGeneration.from_pretrained("gsarti/it5-large")
55
  ```
56
 
57
  *Note: You will need to fine-tune the model on your downstream seq2seq task to use it. See an example [here](https://huggingface.co/gsarti/it5-base-nli).*
48
  ## Using the models
49
 
50
  ```python
51
+ from transformers import AutoTokenzier, AutoModelForSeq2SeqLM
52
 
53
+ tokenizer = AutoTokenizer.from_pretrained("gsarti/it5-large")
54
+ model = AutoModelForSeq2SeqLM.from_pretrained("gsarti/it5-large")
55
  ```
56
 
57
  *Note: You will need to fine-tune the model on your downstream seq2seq task to use it. See an example [here](https://huggingface.co/gsarti/it5-base-nli).*