detector_benchmark.watermark.exp

Submodules

• detector_benchmark.watermark.exp.exp

Classes

EXP	Top-level class for the EXP algorithm.

Package Contents

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

Bases: detector_benchmark.watermark.base.BaseWatermark

Top-level class for the EXP algorithm.

config

utils

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

Generate watermarked text using the EXP algorithm.

generate(encoded_prompts: list, *args, **kwargs) → str

Generate watermarked text. Takes a list of encoded prompts as input, like transformers model.generate.

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

Detect watermark in the text.