diff --git a/src/maxtext/integration/vllm/maxtext_vllm_rollout.py b/src/maxtext/integration/vllm/maxtext_vllm_rollout.py index 54778557e7..9dc530a82b 100644 --- a/src/maxtext/integration/vllm/maxtext_vllm_rollout.py +++ b/src/maxtext/integration/vllm/maxtext_vllm_rollout.py @@ -450,7 +450,7 @@ def __init__( scan_axis: int = 1, layer_pattern_length: Optional[int] = None, ): - super().__init__(tokenizer=tokenizer, config=config, converter=converter) + super().__init__(tokenizer=tokenizer, config=config) self._converter = converter self.converter = converter self._direct_maxtext_sync = direct_maxtext_sync