detector_benchmark.watermark.dip

Submodules

• detector_benchmark.watermark.dip.dip

Classes

DIP	Top-level class for DIP algorithm.

Package Contents

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

Bases: detector_benchmark.watermark.base.BaseWatermark

Top-level class for DIP algorithm.

config

utils

logits_processor

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

Generate watermarked text.

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)

Detect watermark in the text.