Skip to content

Update decoding notebook#156

Merged
SimJeg merged 2 commits intomainfrom
simon/update-decoding-notebook
Nov 13, 2025
Merged

Update decoding notebook#156
SimJeg merged 2 commits intomainfrom
simon/update-decoding-notebook

Conversation

@SimJeg
Copy link
Copy Markdown
Collaborator

@SimJeg SimJeg commented Nov 13, 2025

Update the notebook kvpress_decoding_aime25.ipynb to remove the displays of the cache size which are incorrect as the kvpress pipeline removes the tokens associated with the answer after decoding.

cc @YunSunny00

@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented Nov 13, 2025

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@SimJeg SimJeg force-pushed the simon/update-decoding-notebook branch from ef5009c to cdb4257 Compare November 13, 2025 09:58
@SimJeg SimJeg requested a review from maxjeblick November 13, 2025 10:42
Copy link
Copy Markdown
Collaborator

@maxjeblick maxjeblick left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM thanks!
As discussed, let's update to 0.4

@SimJeg SimJeg force-pushed the simon/update-decoding-notebook branch from 72eb02d to 883ea8d Compare November 13, 2025 13:45
Signed-off-by: SimJeg <sjegou@nvidia.com>
Signed-off-by: SimJeg <sjegou@nvidia.com>
@SimJeg SimJeg force-pushed the simon/update-decoding-notebook branch from 883ea8d to ad2bcb0 Compare November 13, 2025 13:46
@SimJeg SimJeg merged commit 4d3c265 into main Nov 13, 2025
2 checks passed
@SimJeg SimJeg deleted the simon/update-decoding-notebook branch November 13, 2025 13:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants