`watermarking` was refactored to `watermark`

demo_watermark.py

@@ -226,11 +226,11 @@ def generate_with_api(prompt, args):
         generation_params["temperature"] = args.sampling_temp
         generation_params["seed"] = args.generation_seed
 
-    generation_params["watermarking"] = False
+    generation_params["watermark"] = False
     output = client.generate(prompt, **generation_params)
     output_text_without_watermark = output.generated_text
 
-    generation_params["watermarking"] = True
+    generation_params["watermark"] = True
     output = client.generate(prompt, **generation_params)
     output_text_with_watermark = output.generated_text
 

requirements.txt

@@ -6,4 +6,4 @@ torch
 transformers
 tokenizers
 accelerate
-text-generation==0.1.0
+text-generation>=0.3.0