Luna4444 commited on
Commit
0af6e97
1 Parent(s): 7e46b94

Update generators/model.py

Browse files
Files changed (1) hide show
  1. generators/model.py +1 -1
generators/model.py CHANGED
@@ -10,7 +10,7 @@ prompt = PromptTemplate()
10
  stop_words = ["###", "\n\n", "<br><br>", "The authors: "]
11
 
12
 
13
- def model(type, template, seq_len=255):
14
  train = ''
15
  if type == 'title':
16
  train = prompt.TITLE_TO_ABSTRACST
 
10
  stop_words = ["###", "\n\n", "<br><br>", "The authors: "]
11
 
12
 
13
+ def model(type, template, seq_len=250):
14
  train = ''
15
  if type == 'title':
16
  train = prompt.TITLE_TO_ABSTRACST