kz209 commited on
Commit
fcae4c8
1 Parent(s): 29c23fe
Files changed (1) hide show
  1. utils/model.py +1 -1
utils/model.py CHANGED
@@ -19,7 +19,7 @@ class Model():
19
  def gen(self, content, temp=0.0, max_length=500):
20
  sequences = self.pipeline(
21
  content,
22
- max_length=max_length,
23
  do_sample=False,
24
  temperature=temp,
25
  num_return_sequences=1,
 
19
  def gen(self, content, temp=0.0, max_length=500):
20
  sequences = self.pipeline(
21
  content,
22
+ max_new_tokens=max_length,
23
  do_sample=False,
24
  temperature=temp,
25
  num_return_sequences=1,