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

Supply device to SAEConfigLoadOptions #347

Merged
merged 1 commit into from
Oct 23, 2024

Conversation

callummcdougall
Copy link
Contributor

This is a bug which was introduced last week. The issue seems to be that the SAEConfigLoadOptions class was introduced which contains the device field (default None), and this field wasn't set when device is not None. Before this change, the config was just loaded directly from HuggingFace without overriding its device value (which on HuggingFace presumably is "cuda").

Would be great if this PR could be merged soon if possible, and I can try to add tests for this kind of failure mode afterwards.

@callummcdougall callummcdougall mentioned this pull request Oct 23, 2024
@chanind chanind merged commit 531b1c7 into jbloomAus:main Oct 23, 2024
5 checks passed
@chanind
Copy link
Collaborator

chanind commented Oct 23, 2024

Thanks for finding this! just merged as this is a live bug. If you can add a test that would prevent this regression in the future that would be great as well

@callummcdougall
Copy link
Contributor Author

Yep will add a test on my todo list for this week!

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