OrangeEye commited on
Commit
ac7f2bf
·
1 Parent(s): 2caf4d8

update do_sample=False

Browse files
Files changed (1) hide show
  1. utils.py +1 -0
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