set max_new_tokens to 50 instead of 30
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ if __name__ == "__main__":
|
|
43 |
grammar_processor = GrammarConstrainedLogitsProcessor(grammar)
|
44 |
|
45 |
outputs = model.generate(
|
46 |
-
**inputs, max_new_tokens=
|
47 |
)
|
48 |
# Important: don't forget to set `normalize_logits=True` to obtain normalized probabilities (i.e. sum(p) = 1)
|
49 |
transition_scores = model.compute_transition_scores(outputs.sequences, outputs.scores, normalize_logits=True)
|
|
|
43 |
grammar_processor = GrammarConstrainedLogitsProcessor(grammar)
|
44 |
|
45 |
outputs = model.generate(
|
46 |
+
**inputs, max_new_tokens=50, repetition_penalty=1, return_dict_in_generate=True, output_scores=True, logits_processor=[grammar_processor]
|
47 |
)
|
48 |
# Important: don't forget to set `normalize_logits=True` to obtain normalized probabilities (i.e. sum(p) = 1)
|
49 |
transition_scores = model.compute_transition_scores(outputs.sequences, outputs.scores, normalize_logits=True)
|