test recover

demo_watermark.py

@@ -271,7 +271,7 @@ def check_prompt(prompt, args, tokenizer, model=None, device=None):
             int(truncation_warning),
             args)
 
-
+@spaces.GPU
 def generate(prompt, args, tokenizer, model=None, device=None):
     """根据水印参数实例化 WatermarkLogitsProcessor 并通过将其作为 logits 处理器传递给模型的 generate 方法来生成带水印的文本。"""
     print(f"Generating with {args}")