Skip to content

Add integration test for spec decode top k = 1#3

Merged
hebiao064 merged 2 commits intomainfrom
add-spec-dec-top-1
Apr 8, 2025
Merged

Add integration test for spec decode top k = 1#3
hebiao064 merged 2 commits intomainfrom
add-spec-dec-top-1

Conversation

@yubofredwang
Copy link
Copy Markdown
Owner

Motivation

Modifications

Checklist

Comment thread test/srt/test_fa3.py Outdated
"--speculative-eagle-topk",
"1",
"--speculative-num-draft-tokens",
"4",
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

use 3, it need to be the same as --speculative-num-steps when we use topk = 1

@hebiao064 hebiao064 merged commit 77d6ef4 into main Apr 8, 2025
0 of 2 checks passed
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