detector_benchmark.watermark.base

Classes

BaseWatermark

Module Contents

class detector_benchmark.watermark.base.BaseWatermark(algorithm_config: str, gen_model, model_config: detector_benchmark.utils.configs.ModelConfig, *args, **kwargs)

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

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

Generate unwatermarked text.

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