cicdatopea commited on
Commit
71be988
·
verified ·
1 Parent(s): 06eadc3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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, ##change this to align with the official usage
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)