:books: fix readme
Browse files
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=
|
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 |
)
|