Skip to content

Commit a2e2a3f

Browse files
committed
Fix typo
1 parent 180e20a commit a2e2a3f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

megatron/core/inference/engines/dynamic_engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -604,7 +604,7 @@ def get_active_sampling_map(self) -> List[Tuple[SamplingParams, List[int]]]:
604604
self.context.paused_request_count : self.context.total_request_count
605605
].tolist()
606606
if self.static_sampling:
607-
return [next(self.requests.values()).sampling_params, active_request_ids]
607+
return [next(iter(self.requests.values())).sampling_params, active_request_ids]
608608

609609
# Get a map from request_id to context array index.
610610
context_id_map = {r: i for i, r in enumerate(active_request_ids)}

0 commit comments

Comments
 (0)