From 8666df0f41788c3386e1f6298276b6db5519d9a1 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Tue, 15 Aug 2023 21:21:57 +0200
Subject: [PATCH] Fixing watermark.

---
 server/text_generation_server/utils/watermark.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/server/text_generation_server/utils/watermark.py b/server/text_generation_server/utils/watermark.py
index df7b90e3..5d8f5312 100644
--- a/server/text_generation_server/utils/watermark.py
+++ b/server/text_generation_server/utils/watermark.py
@@ -19,8 +19,8 @@ import torch
 from transformers import LogitsProcessor
 from typing import List, Union
 
-GAMMA = os.getenv("WATERMARK_GAMMA", 0.5)
-DELTA = os.getenv("WATERMARK_DELTA", 2.0)
+GAMMA = float(os.getenv("WATERMARK_GAMMA", 0.5))
+DELTA = float(os.getenv("WATERMARK_DELTA", 2.0))
 
 
 class WatermarkLogitsProcessor(LogitsProcessor):