Overglitch commited on
Commit
4a25d51
·
verified ·
1 Parent(s): c4a0aa6

Update modules/abstractive.py

Browse files
Files changed (1) hide show
  1. modules/abstractive.py +1 -1
modules/abstractive.py CHANGED
@@ -24,7 +24,7 @@ def load_summarizers():
24
  def abstractive_summary(summarizers, model_name, text, max_length, num_beams):
25
  model, tokenizer = summarizers[model_name]
26
  inputs = tokenizer(
27
- text, return_tensors="pt", max_length=512, truncation=True
28
  ).to(model.device)
29
  outputs = model.generate(
30
  inputs["input_ids"],
 
24
  def abstractive_summary(summarizers, model_name, text, max_length, num_beams):
25
  model, tokenizer = summarizers[model_name]
26
  inputs = tokenizer(
27
+ text, return_tensors="pt", max_length=1024, truncation=True
28
  ).to(model.device)
29
  outputs = model.generate(
30
  inputs["input_ids"],