Spaces:
Running
Running
increase long t5 max_input_length
Browse files
app.py
CHANGED
@@ -66,7 +66,7 @@ def proc_submission(
|
|
66 |
st = time.perf_counter()
|
67 |
history = {}
|
68 |
clean_text = clean(input_text, lower=False)
|
69 |
-
max_input_length =
|
70 |
processed = truncate_word_count(clean_text, max_input_length)
|
71 |
|
72 |
if processed["was_truncated"]:
|
|
|
66 |
st = time.perf_counter()
|
67 |
history = {}
|
68 |
clean_text = clean(input_text, lower=False)
|
69 |
+
max_input_length = 2048 if "base" in model_size.lower() else max_input_length
|
70 |
processed = truncate_word_count(clean_text, max_input_length)
|
71 |
|
72 |
if processed["was_truncated"]:
|