enable greedy
#1
by
merve
HF staff
- opened
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=
|
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
|