Spaces:
Sleeping
Sleeping
♻️ generalize size
Browse filesSigned-off-by: peter szemraj <peterszemraj@gmail.com>
app.py
CHANGED
@@ -82,8 +82,8 @@ def proc_submission(
|
|
82 |
|
83 |
_summaries = summarize_via_tokenbatches(
|
84 |
tr_in,
|
85 |
-
model_sm if
|
86 |
-
tokenizer_sm if
|
87 |
batch_length=token_batch_length,
|
88 |
**settings,
|
89 |
)
|
|
|
82 |
|
83 |
_summaries = summarize_via_tokenbatches(
|
84 |
tr_in,
|
85 |
+
model_sm if "base" in model_size.lower() else model,
|
86 |
+
tokenizer_sm if "base" in model_size.lower() else tokenizer,
|
87 |
batch_length=token_batch_length,
|
88 |
**settings,
|
89 |
)
|