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

Add weight tying and input scaling to MLPSpeculator #36

Merged
merged 23 commits into from
Jul 8, 2024

Conversation

sahilsuneja1
Copy link
Contributor

... and corresponding options to MLPSpeculatorConfig to create the HF version. Builds on top of @daviswer's PR #25

@sahilsuneja1
Copy link
Contributor Author

@JRosenkranz @daviswer Can we please get this merged-- needed for supporting speculators in watsonx

Copy link
Collaborator

@JRosenkranz JRosenkranz left a comment

Choose a reason for hiding this comment

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

Mostly looks good to me, just need to check the tie_wts parameter

@sahilsuneja1
Copy link
Contributor Author

@JRosenkranz updated to use only 1 flag now-- tie_weights as per our discussion

@JRosenkranz JRosenkranz self-requested a review July 8, 2024 18:47
Copy link
Collaborator

@JRosenkranz JRosenkranz left a comment

Choose a reason for hiding this comment

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

lgtm

@JRosenkranz JRosenkranz merged commit 50ac056 into foundation-model-stack:main Jul 8, 2024
3 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.

3 participants