Spaces:
Runtime error
Runtime error
Commit
β’
65966af
1
Parent(s):
6893a8c
fix length
Browse files
app.py
CHANGED
@@ -146,7 +146,7 @@ def model_inference(
|
|
146 |
generation_kwargs = dict(
|
147 |
inputs,
|
148 |
bad_words_ids=BAD_WORDS_IDS,
|
149 |
-
max_length=
|
150 |
)
|
151 |
# Regular generation version
|
152 |
generated_ids = MODEL.generate(**generation_kwargs)
|
|
|
146 |
generation_kwargs = dict(
|
147 |
inputs,
|
148 |
bad_words_ids=BAD_WORDS_IDS,
|
149 |
+
max_length=128,
|
150 |
)
|
151 |
# Regular generation version
|
152 |
generated_ids = MODEL.generate(**generation_kwargs)
|