cicdatopea
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -82,8 +82,8 @@ outputs = model.generate(
|
|
82 |
input_ids=inputs["input_ids"].to(model.device),
|
83 |
attention_mask=inputs["attention_mask"].to(model.device),
|
84 |
max_length=200, ##change this to align with the official usage
|
85 |
-
num_return_sequences=1,
|
86 |
-
do_sample=False
|
87 |
)
|
88 |
generated_ids = [
|
89 |
output_ids[len(input_ids):] for input_ids, output_ids in zip(inputs["input_ids"], outputs)
|
|
|
82 |
input_ids=inputs["input_ids"].to(model.device),
|
83 |
attention_mask=inputs["attention_mask"].to(model.device),
|
84 |
max_length=200, ##change this to align with the official usage
|
85 |
+
num_return_sequences=1,
|
86 |
+
do_sample=False ##change this to align with the official usage
|
87 |
)
|
88 |
generated_ids = [
|
89 |
output_ids[len(input_ids):] for input_ids, output_ids in zip(inputs["input_ids"], outputs)
|