Quintino Fernandes commited on
Commit
542722d
·
1 Parent(s): 4902a85

Model parameters change

Browse files
Files changed (1) hide show
  1. models/summarization.py +6 -6
models/summarization.py CHANGED
@@ -12,18 +12,18 @@ class SummarizationModel:
12
  # Model and tokenization parameters
13
  inputs = self.tokenizer.encode(
14
  text,
15
- max_length=512,
16
  truncation=True,
17
  return_tensors='pt'
18
  ).to(self.device)
19
 
20
  summary_ids = self.model.generate(
21
  inputs,
22
- max_length=256,
23
- min_length=64,
24
- num_beams=4,
25
- no_repeat_ngram_size=3,
26
- early_stopping=True,
27
  )
28
 
29
  return self.tokenizer.decode(summary_ids[0], skip_special_tokens=True)
 
12
  # Model and tokenization parameters
13
  inputs = self.tokenizer.encode(
14
  text,
15
+ max_length=1024,
16
  truncation=True,
17
  return_tensors='pt'
18
  ).to(self.device)
19
 
20
  summary_ids = self.model.generate(
21
  inputs,
22
+ max_length=512,
23
+ min_length=128,
24
+ num_beams=3,
25
+ no_repeat_ngram_size=4,
26
+ early_stopping=False,
27
  )
28
 
29
  return self.tokenizer.decode(summary_ids[0], skip_special_tokens=True)