detector_benchmark.watermark.unbiased

Submodules

• detector_benchmark.watermark.unbiased.unbiased

Classes

UnbiasedWatermark

Top-level class for Unbiased algorithm.

Package Contents

class detector_benchmark.watermark.unbiased.UnbiasedWatermark(algorithm_config: dict, gen_model, transformers_config: detector_benchmark.utils.configs.ModelConfig, *args, **kwargs)

Bases: detector_benchmark.watermark.base.BaseWatermark

Top-level class for Unbiased algorithm.

config

utils

logits_processor

generate_watermarked_text(prompt: str, *args, **kwargs) → str

Generate watermarked text.

detect_watermark(text: str, return_dict: bool = True, *args, **kwargs)

Detect watermark in the text.