Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -38,7 +38,7 @@ def extend(input_text, num_return_sequences, bad_words, max_size=20, top_k=50, t
|
|
| 38 |
input_ids=input_ids,
|
| 39 |
max_length=max_size + len(encoded_prompt[0]),
|
| 40 |
top_k=top_k,
|
| 41 |
-
|
| 42 |
top_p=top_p,
|
| 43 |
do_sample=True,
|
| 44 |
num_return_sequences=num_return_sequences)
|
|
|
|
| 38 |
input_ids=input_ids,
|
| 39 |
max_length=max_size + len(encoded_prompt[0]),
|
| 40 |
top_k=top_k,
|
| 41 |
+
bad_words_ids = bad_word_ids,
|
| 42 |
top_p=top_p,
|
| 43 |
do_sample=True,
|
| 44 |
num_return_sequences=num_return_sequences)
|