[Bugfix] Migrate to REGEX Library to prevent catastrophic backtracking (#18454)

Signed-off-by: Crucifixion-Fxl <xmufxl@gmail.com> Co-authored-by: Crucifixion-Fxl <xmufxl@gmail.com>
2025-05-24 07:16:26 +08:00
parent f2036734fb
commit 4fc1bf813a
52 changed files with 62 additions and 58 deletions
--- a/vllm/model_executor/models/mimo_mtp.py
+++ b/vllm/model_executor/models/mimo_mtp.py
@@ -250,7 +250,7 @@ class MiMoMTP(nn.Module):
        return loaded_params

    def map_model_name_to_mtp_param_name(self, name: str) -> str:
-        import re
+        import regex as re
        name_without_prefix = [
            "token_layernorm", "hidden_layernorm", "input_proj",
            "final_layernorm"