From a966dd62f74068f560b6662db331471b39e9e20c Mon Sep 17 00:00:00 2001 From: r1chardf1d0 Date: Mon, 21 Jul 2025 21:18:50 +0800 Subject: [PATCH] fix profile and chat --- vllm_mindspore/__init__.py | 7 ------- vllm_mindspore/worker/profile.py | 9 +++++++++ 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/vllm_mindspore/__init__.py b/vllm_mindspore/__init__.py index a63a06a74..a4e049379 100644 --- a/vllm_mindspore/__init__.py +++ b/vllm_mindspore/__init__.py @@ -464,13 +464,6 @@ import vllm.engine.multiprocessing.engine vllm.engine.multiprocessing.engine.MQLLMEngine.cleanup = cleanup -from vllm.entrypoints.openai.serving_chat import OpenAIServingChat -from vllm_mindspore.entrypoints.openai.serving_chat import ( - chat_completion_stream_generator, ) - -OpenAIServingChat.chat_completion_stream_generator = ( - chat_completion_stream_generator) - from vllm_mindspore.entrypoints.openai.tool_parsers import ( deepseekv3_tool_parser, ) diff --git a/vllm_mindspore/worker/profile.py b/vllm_mindspore/worker/profile.py index c99059db5..0a35f81ba 100644 --- a/vllm_mindspore/worker/profile.py +++ b/vllm_mindspore/worker/profile.py @@ -54,6 +54,15 @@ class AdapterProfiler: path = ProfilerContext().ascend_ms_dir shell_analyse(path) + def key_averages(self): + + class _inner: + + def table(self, sort_by=None): + return "" + + return _inner() + def wrapper_worker_init(fun): -- Gitee