enable greedy

#1
by merve HF staff - opened
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -104,10 +104,9 @@ inputs = {key: value.to(device) for (key, value) in inputs.items()}
104
  # Generate the output
105
  output = model.generate(
106
  **inputs,
107
- temperature=0.8,
108
  max_new_tokens=50,
109
  num_return_sequences=1,
110
- do_sample=True,
111
  )
112
 
113
  # Decode the output
 
104
  # Generate the output
105
  output = model.generate(
106
  **inputs,
 
107
  max_new_tokens=50,
108
  num_return_sequences=1,
109
+ do_sample=False,
110
  )
111
 
112
  # Decode the output