asi commited on
Commit
ea1eeed
1 Parent(s): 03bc643

:books: fix readme

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -46,10 +46,9 @@ input_ids = tokenizer.encode(input_sentence, return_tensors='pt')
46
 
47
  beam_outputs = model.generate(
48
  input_ids,
49
- max_length=200,
50
  do_sample=True,
51
  top_k=50,
52
- max_length=100,
53
  top_p=0.95,
54
  num_return_sequences=1
55
  )
 
46
 
47
  beam_outputs = model.generate(
48
  input_ids,
49
+ max_length=100,
50
  do_sample=True,
51
  top_k=50,
 
52
  top_p=0.95,
53
  num_return_sequences=1
54
  )