Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: update verify process #47

Merged
merged 1 commit into from
Mar 2, 2024
Merged

fix: update verify process #47

merged 1 commit into from
Mar 2, 2024

Conversation

cyLi-Tiger
Copy link
Contributor

@cyLi-Tiger cyLi-Tiger commented Feb 29, 2024

I think the verification process needs another condition. According to the prove in SpecInfer, when all candidates in one draft candidate path are accepted, we need to sample the sample_token from the logits of the last accepted node. Here, you still sample from the gtp and the last accepted token is also sampled from gtp, which may cause repetition in the output. Here is what I got from your example.

image

Please correct me if I misunderstood anything!

@Liyuhui-12
Copy link
Collaborator

Thank you.

@Liyuhui-12 Liyuhui-12 merged commit 868b2f5 into SafeAILab:main Mar 2, 2024
@Liyuhui-12 Liyuhui-12 mentioned this pull request Mar 7, 2024
JoursBleu pushed a commit to JoursBleu/EAGLE that referenced this pull request Jun 28, 2024
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