Update README.md
Browse files
README.md
CHANGED
@@ -24,7 +24,7 @@ inp = tokenizer('Wen dey wen stretch him out fo whip him real hard , Paul wen te
|
|
24 |
|
25 |
model.eval() # Set dropouts to zero
|
26 |
|
27 |
-
model_output=model.generate(inp, use_cache=True, num_beams=4, max_length=60, min_length=1, early_stopping=True, pad_token_id=pad_id, bos_token_id=bos_id, eos_token_id=eos_id, decoder_start_token_id=tokenizer._convert_token_to_id_with_added_voc("<
|
28 |
|
29 |
|
30 |
# Decode to get output strings
|
|
|
24 |
|
25 |
model.eval() # Set dropouts to zero
|
26 |
|
27 |
+
model_output=model.generate(inp, use_cache=True, num_beams=4, max_length=60, min_length=1, early_stopping=True, pad_token_id=pad_id, bos_token_id=bos_id, eos_token_id=eos_id, decoder_start_token_id=tokenizer._convert_token_to_id_with_added_voc("<2eng>"))
|
28 |
|
29 |
|
30 |
# Decode to get output strings
|