Update README.md
Browse files
README.md
CHANGED
@@ -45,7 +45,7 @@ model_output=model.generate(inp, use_cache=True, num_beams=4, max_length=20, min
|
|
45 |
decoded_output=tokenizer.decode(model_output[0], skip_special_tokens=True, clean_up_tokenization_spaces=False)
|
46 |
|
47 |
print(decoded_output) # I am a boy
|
48 |
-
|
49 |
Notes:
|
50 |
1. This is compatible with the latest version of transformers but was developed with version 4.3.2 so consider using 4.3.2 if possible.
|
51 |
2. While I have only shown how to let logits and loss and how to generate outputs, you can do pretty much everything the MBartForConditionalGeneration class can do as in https://huggingface.co/docs/transformers/model_doc/mbart#transformers.MBartForConditionalGeneration
|
|
|
45 |
decoded_output=tokenizer.decode(model_output[0], skip_special_tokens=True, clean_up_tokenization_spaces=False)
|
46 |
|
47 |
print(decoded_output) # I am a boy
|
48 |
+
```
|
49 |
Notes:
|
50 |
1. This is compatible with the latest version of transformers but was developed with version 4.3.2 so consider using 4.3.2 if possible.
|
51 |
2. While I have only shown how to let logits and loss and how to generate outputs, you can do pretty much everything the MBartForConditionalGeneration class can do as in https://huggingface.co/docs/transformers/model_doc/mbart#transformers.MBartForConditionalGeneration
|