Spaces:
Runtime error
Runtime error
update do_sample=False
Browse files
utils.py
CHANGED
|
@@ -179,6 +179,7 @@ def moderate(chat, model, tokenizer, UNSAFE_TOKEN_ID):
|
|
| 179 |
return_dict_in_generate=True,
|
| 180 |
pad_token_id=tokenizer.eos_token_id,
|
| 181 |
output_logits=True, # get logits
|
|
|
|
| 182 |
)
|
| 183 |
######
|
| 184 |
# Get generated text
|
|
|
|
| 179 |
return_dict_in_generate=True,
|
| 180 |
pad_token_id=tokenizer.eos_token_id,
|
| 181 |
output_logits=True, # get logits
|
| 182 |
+
do_sample=False
|
| 183 |
)
|
| 184 |
######
|
| 185 |
# Get generated text
|