pszemraj commited on
Commit
0fccfa2
1 Parent(s): acfdb35

increase long t5 max_input_length

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = 1536 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"]:
 
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"]: