Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -67,6 +67,7 @@ def model_inference(
|
|
| 67 |
generation_args["temperature"] = temperature
|
| 68 |
generation_args["do_sample"] = True
|
| 69 |
generation_args["top_p"] = top_p
|
|
|
|
| 70 |
|
| 71 |
|
| 72 |
generation_args.update(inputs)
|
|
|
|
| 67 |
generation_args["temperature"] = temperature
|
| 68 |
generation_args["do_sample"] = True
|
| 69 |
generation_args["top_p"] = top_p
|
| 70 |
+
generation_args["bad_words_ids"] = BAD_WORDS_IDS
|
| 71 |
|
| 72 |
|
| 73 |
generation_args.update(inputs)
|