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 bleeding edge model loads and add lazyload fallback #390

Merged
merged 3 commits into from
Jul 8, 2023

Conversation

one-some
Copy link

@one-some one-some commented Jul 8, 2023

This PR:

  • Fixes _rebuild_from_type_v2 shenanigans that occur on torch 2.0
  • Adds the --panic argument to raise all errors instead of attempting to fall back (for debugging)
  • Attempts to fall back to unpatched HF loading if lazyloading fails.

@one-some one-some changed the title Accelerate offloading Fix bleeding edge model loads and add lazyload fallback Jul 8, 2023
@one-some one-some mentioned this pull request Jul 8, 2023
@henk717 henk717 merged commit 95913ae into henk717:united Jul 8, 2023
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