Skip to content

Commit df5afa8

Browse files
authored
[Log] Debug Once for Randomizing dummy data for DP Rank (#22860)
Signed-off-by: yewentao256 <[email protected]>
1 parent 6cd69f5 commit df5afa8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/v1/worker/gpu_model_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2192,7 +2192,7 @@ def rand_input_ids() -> torch.Tensor:
21922192
high=self.model_config.get_vocab_size(),
21932193
dtype=input_ids.dtype)
21942194

2195-
logger.debug("Randomizing dummy data for DP Rank")
2195+
logger.debug_once("Randomizing dummy data for DP Rank")
21962196
input_ids.copy_(rand_input_ids()[:input_ids.size(0)],
21972197
non_blocking=True)
21982198
yield

0 commit comments

Comments
 (0)