Spaces:
Running
on
Zero
Running
on
Zero
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
|