Spaces:
Sleeping
Sleeping
Quintino Fernandes
commited on
Commit
·
542722d
1
Parent(s):
4902a85
Model parameters change
Browse files- 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=
|
| 16 |
truncation=True,
|
| 17 |
return_tensors='pt'
|
| 18 |
).to(self.device)
|
| 19 |
|
| 20 |
summary_ids = self.model.generate(
|
| 21 |
inputs,
|
| 22 |
-
max_length=
|
| 23 |
-
min_length=
|
| 24 |
-
num_beams=
|
| 25 |
-
no_repeat_ngram_size=
|
| 26 |
-
early_stopping=
|
| 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)
|