Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions vllm/v1/worker/gpu_model_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -5664,6 +5664,14 @@ def _dummy_sampler_run(
sampler_output = self.sampler(
logits=logits, sampling_metadata=dummy_metadata
)
# Also warm forward_native (taken when generators dict is non-empty).
self.sampler(
logits=logits,
sampling_metadata=replace(
dummy_metadata,
generators={0: torch.Generator(device=self.device).manual_seed(0)},
),
)
Comment thread
arpera marked this conversation as resolved.
except RuntimeError as e:
if "out of memory" in str(e):
raise RuntimeError(
Expand Down
Loading