[Log] Debug Once for Randomizing dummy data for DP Rank (#22860)
Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
@ -2192,7 +2192,7 @@ class GPUModelRunner(LoRAModelRunnerMixin, KVConnectorModelRunnerMixin):
|
||||
high=self.model_config.get_vocab_size(),
|
||||
dtype=input_ids.dtype)
|
||||
|
||||
logger.debug("Randomizing dummy data for DP Rank")
|
||||
logger.debug_once("Randomizing dummy data for DP Rank")
|
||||
input_ids.copy_(rand_input_ids()[:input_ids.size(0)],
|
||||
non_blocking=True)
|
||||
yield
|
||||
|
||||
Reference in New Issue
Block a user