Spaces:
Sleeping
Sleeping
kz209
commited on
Commit
•
fcae4c8
1
Parent(s):
29c23fe
update
Browse files- 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 |
-
|
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,
|