prajdabre commited on
Commit
41abcd0
1 Parent(s): 8150963

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -35,7 +35,7 @@ model_outputs.logits
35
 
36
  # For generation. Pardon the messiness. Note the decoder_start_token_id.
37
 
38
- model.eval() # Det dropouts to zero
39
 
40
  model_output=model.generate(inp, use_cache=True, num_beams=4, max_length=20, 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("<2en>"))
41
 
 
35
 
36
  # For generation. Pardon the messiness. Note the decoder_start_token_id.
37
 
38
+ model.eval() # Set dropouts to zero
39
 
40
  model_output=model.generate(inp, use_cache=True, num_beams=4, max_length=20, 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("<2en>"))
41