codable-watermarking-for-llm
codable-watermarking-for-llm copied to clipboard
Bug in the implementation when non_analyze=False
In watermarking/watermark_processors/message_model_processor.py
decoded_message, decoded_prob = decoded_probs.max(0)
should be decoded_prob, decoded_message = decoded_probs.max(0)