vesteinn commited on
Commit
d0e3a42
1 Parent(s): 1cb7e87

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -21,7 +21,7 @@ It will occasionally combine sentences or add an extra sentence.
21
 
22
  This is the same model as are provided on CLARIN: https://repository.clarin.is/repository/xmlui/handle/20.500.12537/278
23
 
24
- You can use the following example to get started (note that it is necessary to alter the `decoder_start_token_id` of the model):
25
 
26
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
27
  import torch
@@ -31,7 +31,6 @@ You can use the following example to get started (note that it is necessary to a
31
  tokenizer = AutoTokenizer.from_pretrained("mideind/nmt-doc-en-is-2022-10",src_lang="en_XX",tgt_lang="is_IS")
32
 
33
  model = AutoModelForSeq2SeqLM.from_pretrained("mideind/nmt-doc-en-is-2022-10")
34
- model.config.decoder_start_token_id = 2
35
 
36
  translate = pipeline("translation_XX_to_YY",model=model,tokenizer=tokenizer,device=device,src_lang="en_XX",tgt_lang="is_IS")
37
 
21
 
22
  This is the same model as are provided on CLARIN: https://repository.clarin.is/repository/xmlui/handle/20.500.12537/278
23
 
24
+ You can use the following example to get started:
25
 
26
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
27
  import torch
31
  tokenizer = AutoTokenizer.from_pretrained("mideind/nmt-doc-en-is-2022-10",src_lang="en_XX",tgt_lang="is_IS")
32
 
33
  model = AutoModelForSeq2SeqLM.from_pretrained("mideind/nmt-doc-en-is-2022-10")
 
34
 
35
  translate = pipeline("translation_XX_to_YY",model=model,tokenizer=tokenizer,device=device,src_lang="en_XX",tgt_lang="is_IS")
36