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

script to estimate qlora mem usage #1294

Conversation

Titus-von-Koeller
Copy link
Collaborator

sketch of a script to estimate qlora mem usage (w/o taking FSDP into account)

@AaronZLT
Copy link

AaronZLT commented Aug 4, 2024

Could you explain the calculation in the estimation?

@Titus-von-Koeller
Copy link
Collaborator Author

@AaronZLT Sry, this was just a draft based on a formula that Tim came up with, but it didn't really prove to be useful. I don't have a good explanation off the top of my head right now and need to focus on other tasks right now, sry!

Closing this due to the above reason.

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