7ee8a6dd01
Signed-off-by: Raphael Rialland <raphael.rialland@mistral.ai> Signed-off-by: Simon Veitner <sveitner@redhat.com> Signed-off-by: Tomas Ruiz <tomas.ruiz.te@gmail.com> Co-authored-by: OpenAI Codex <codex@openai.com> Co-authored-by: Simon Veitner <sveitner@redhat.com> Co-authored-by: Tomas Ruiz <tomas.ruiz.te@gmail.com>
40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from vllm.config.watermarking import derive_watermark_key
|
|
from vllm.v1.watermarking.detector import WatermarkDetection, WatermarkDetector
|
|
from vllm.v1.watermarking.factory import create_watermarker
|
|
from vllm.v1.watermarking.gumbel import (
|
|
DualKeyGumbelWatermarkDetector,
|
|
DualKeyGumbelWatermarker,
|
|
GumbelWatermarkDetector,
|
|
GumbelWatermarker,
|
|
)
|
|
from vllm.v1.watermarking.prfs import (
|
|
PhiloxPRF,
|
|
WatermarkPRF,
|
|
create_prf,
|
|
)
|
|
from vllm.v1.watermarking.watermarker import (
|
|
SupportsSpeculativeDecoding,
|
|
Watermarker,
|
|
WatermarkSample,
|
|
)
|
|
|
|
__all__ = [
|
|
"DualKeyGumbelWatermarkDetector",
|
|
"DualKeyGumbelWatermarker",
|
|
"GumbelWatermarkDetector",
|
|
"GumbelWatermarker",
|
|
"PhiloxPRF",
|
|
"SupportsSpeculativeDecoding",
|
|
"WatermarkDetection",
|
|
"WatermarkDetector",
|
|
"WatermarkPRF",
|
|
"Watermarker",
|
|
"WatermarkSample",
|
|
"create_watermarker",
|
|
"create_prf",
|
|
"derive_watermark_key",
|
|
]
|