pszemraj commited on
Commit
d5e892e
β€’
1 Parent(s): d378ba6

πŸ› max input len now longer for base

Browse files

Signed-off-by: peter szemraj <[email protected]>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ def proc_submission(
65
  st = time.perf_counter()
66
  history = {}
67
  clean_text = clean(input_text, lower=False)
68
- max_input_length = 2048 if model_size == "base" else max_input_length
69
  processed = truncate_word_count(clean_text, max_input_length)
70
 
71
  if processed["was_truncated"]:
 
65
  st = time.perf_counter()
66
  history = {}
67
  clean_text = clean(input_text, lower=False)
68
+ max_input_length = 2560 if "base" in model_size.lower() else max_input_length
69
  processed = truncate_word_count(clean_text, max_input_length)
70
 
71
  if processed["was_truncated"]: