Spaces:
Runtime error
Runtime error
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,
|