[BugFix] fix ErnieProcessor not set raw_prediction (#3400)

2025-10-05 08:37:06 +08:00 · 2025-08-14 18:07:49 +08:00
parent f72db9386c
commit 6031f9a5f5
2 changed files with 54 additions and 0 deletions
--- a/fastdeploy/input/ernie_processor.py
+++ b/fastdeploy/input/ernie_processor.py
@@ -287,6 +287,7 @@ class ErnieProcessor(BaseDataProcessor):
            if token_ids[-1] == self.tokenizer.eos_token_id:
                token_ids = token_ids[:-1]
        delta_text, previous_token_ids, previous_texts = self.ids2tokens(token_ids, req_id)
+        response_dict["outputs"]["raw_prediction"] = delta_text
        if self.reasoning_parser and (
            enable_thinking or self.reasoning_parser.__class__.__name__ == "ErnieX1ReasoningParser"
        ):