xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-03-24 08:21:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Chore] adjust log info

Browse Source
This commit is contained in:
i-yuanyukun 2025-12-19 16:04:47 +08:00
parent 65ea10c8f4
commit bde36017fa
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
vllm/v1/worker/gpu_ffn_model_runner.py
View File

@ -137,10 +137,10 @@ class GPUFFNModelRunner(LoRAModelRunnerMixin):
else:
dp_metadata = None
current_layer_idx = recv_metadata.layer_idx
logger.info(
f"layer {current_layer_idx} moe recv hidden states type:{type(hidden_states)}, shape:{hidden_states.shape}"
f" dp_metadata: {dp_metadata}"
)
# logger.info(
# f"layer {current_layer_idx} moe recv hidden states type:{type(hidden_states)}, shape:{hidden_states.shape}"
# f" dp_metadata: {dp_metadata}"
# )
num_tokens = hidden_states.shape[0]
if recv_metadata is not None and recv_metadata.recv_handle_list is not None:
for work in recv_metadata.recv_handle_list: