🎨 Remove new_token_ids from warmup

[prashantgupta24]

Description

🎨 Remove new_token_ids from warmup since new_token_ids are not used anymore.

[github-actions]

👋 Hi! Thank you for contributing to vLLM support on Spyre.
Just a reminder: Make sure that your code passes all the linting checks, otherwise your PR won't be able to be merged. To do so, first install the linting requirements, then run format.sh and commit the changes. This can be done with uv directly:

uv sync --frozen --group lint --active --inexact

Or this can be done with pip:

uv pip compile --group lint > requirements-lint.txt
pip install -r requirements-lint.txt
bash format.sh

Now you are good to go 🚀

[prashantgupta24]

bot:test
MARKERS="cb and spyre"

[joerunde]

@prashantgupta24 do we wanna get this merged?

If so we should definitely test to triple-check that this works with the upcoming compiler changes for continuous batching

[prashantgupta24]

No hurry as such, don't want to add anything before the release lol

[prashantgupta24]

bot:test
MARKERS="spyre and not quantized"

[prashantgupta24]

bot:test
MARKERS="spyre and not quantized"

[maxdebayser]

LGTM

[prashantgupta24]

Failed one of the tests on spyre

tests/e2e/test_spyre_cb_scheduler_steps.py::test_new_sequence_joins_during_decode[sendnn-ibm-ai-platform/micro-g3.3-8b-instruct-1b] FAILED

will investigate, seems flaky...

[prashantgupta24]

bot:test
MARKERS="spyre and cb" - all cb tests passed ! (170)

[prashantgupta24]

bot:test
MARKERS="spyre and not quantized"

[prashantgupta24]

Failed 2 different tests this time -

tests/e2e/test_spyre_cb_scheduler_steps.py::test_prompt_too_long_for_current_tkv[sendnn-ibm-ai-platform/micro-g3.3-8b-instruct-1b] FAILED
tests/e2e/test_spyre_cb_scheduler_steps.py::test_requested_tokens_not_fitting_remaining_space[sendnn-ibm-ai-platform/micro-g3.3-8b-instruct-1b] FAILED

Seems flaky since running only CB tests made them all pass 🤔

[prashantgupta24]

Too much flakiness going on, going to convert it to a draft for now.

[prashantgupta24]

bot:test