update

utils.py

@@ -181,6 +181,7 @@ def moderate(chat, model, tokenizer, UNSAFE_TOKEN_ID):
         return_dict_in_generate=True,
         pad_token_id=tokenizer.eos_token_id,
         output_logits=True,  # get logits
+        do_sample=False,
     )
     ######
     # Get generated text