Update README.md
Browse files
README.md
CHANGED
@@ -30,7 +30,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained("tartuNLP/m2m100_418M_smugri")
|
|
30 |
|
31 |
tokenizer.src_lang = 'liv'
|
32 |
|
33 |
-
encoded_src = tokenizer("Līvõ kēļ jelāb
|
34 |
|
35 |
encoded_out = model.generate(**encoded_src, forced_bos_token_id = tokenizer.get_lang_id("sme"))
|
36 |
print(tokenizer.batch_decode(encoded_out, skip_special_tokens=True))
|
|
|
30 |
|
31 |
tokenizer.src_lang = 'liv'
|
32 |
|
33 |
+
encoded_src = tokenizer("Līvõ kēļ jelāb!", return_tensors="pt")
|
34 |
|
35 |
encoded_out = model.generate(**encoded_src, forced_bos_token_id = tokenizer.get_lang_id("sme"))
|
36 |
print(tokenizer.batch_decode(encoded_out, skip_special_tokens=True))
|