detector_benchmark.watermark.synth_id

Submodules

• detector_benchmark.watermark.synth_id.detector
• detector_benchmark.watermark.synth_id.detector_bayesian_torch
• detector_benchmark.watermark.synth_id.synth_id

Classes

SynthID

Top-level class for SynthID algorithm.

Package Contents

class detector_benchmark.watermark.synth_id.SynthID(algorithm_config: str, gen_model, transformers_config: detector_benchmark.utils.configs.ModelConfig, *args, **kwargs)

Bases: detector_benchmark.watermark.base.BaseWatermark

Top-level class for SynthID algorithm.

config

utils

logits_processor

detector

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.

Args:

text (str): Text to detect watermark in return_dict (bool): Whether to return results as dictionary

Returns:

Union[Dict[str, Union[bool, float]], Tuple[bool, float]]: Detection results