xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-07-09 18:47:15 +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: else:
dp_metadata = None dp_metadata = None
current_layer_idx = recv_metadata.layer_idx current_layer_idx = recv_metadata.layer_idx
logger.info( # logger.info(
f"layer {current_layer_idx} moe recv hidden states type:{type(hidden_states)}, shape:{hidden_states.shape}" # f"layer {current_layer_idx} moe recv hidden states type:{type(hidden_states)}, shape:{hidden_states.shape}"
f" dp_metadata: {dp_metadata}" # f" dp_metadata: {dp_metadata}"
) # )
num_tokens = hidden_states.shape[0] num_tokens = hidden_states.shape[0]
if recv_metadata is not None and recv_metadata.recv_handle_list is not None: if recv_metadata is not None and recv_metadata.recv_handle_list is not None:
for work in recv_metadata.recv_handle_list: for work in recv_metadata.recv_handle_list: