tab

handler.py

@@ -87,15 +87,15 @@ class EndpointHandler:
                 return_dict_in_generate=True,
                 output_scores=True
             )
-            # 🧠 打印 debug 信息
-            generated_ids = output.sequences[0]
-            print("📍生成的 token ids：", generated_ids.tolist())
-            raw_text = self.tokenizer.decode(generated_ids, skip_special_tokens=False)
-            clean_text = self.tokenizer.decode(generated_ids, skip_special_tokens=True)
-            print("🧾 带特殊符号的输出：", raw_text)
-            print("✅ 去掉特殊符号的输出：", clean_text)
-
-            return clean_text  # 返回干净版本
+        # 🧠 打印 debug 信息
+        generated_ids = output.sequences[0]
+        print("📍生成的 token ids：", generated_ids.tolist())
+        raw_text = self.tokenizer.decode(generated_ids, skip_special_tokens=False)
+        clean_text = self.tokenizer.decode(generated_ids, skip_special_tokens=True)
+        print("🧾 带特殊符号的输出：", raw_text)
+        print("✅ 去掉特殊符号的输出：", clean_text)
+
+        return clean_text  # 返回干净版本
     def postprocess(self, output: str) -> Dict[str, str]:
         return {"output": output.strip()}