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

Demote error log to warning for models treated as having size 0 #6589

Merged
merged 1 commit into from
Jul 9, 2024

Conversation

RyanJDick
Copy link
Collaborator

@RyanJDick RyanJDick commented Jul 9, 2024

Summary

Demote error log to warning for models treated as having size 0.

Related Issues / Discussions

Closes #6587

I looked into handling ESRGAN model sizes properly. They load a state_dict with a bit of an unusual nested-dict structure. Rather than figure out how to accurately calculate their size, we can just wait for #6556. ESRGAN model size handling should work properly when loaded through that pathway.

QA Instructions

Loaded an ESRGAN model, and confirmed that the warning log is at the warning level.

Merge Plan

No special instructions.

Checklist

  • The PR has a short but descriptive title, suitable for a changelog
  • Tests added / updated (if applicable)
  • Documentation added / updated (if applicable)

@github-actions github-actions bot added python PRs that change python files backend PRs that change backend files labels Jul 9, 2024
@RyanJDick RyanJDick enabled auto-merge July 9, 2024 12:46
@RyanJDick RyanJDick merged commit 9ae8087 into main Jul 9, 2024
14 checks passed
@RyanJDick RyanJDick deleted the ryan/esrgan-model-size branch July 9, 2024 12:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
backend PRs that change backend files python PRs that change python files
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[bug]: ESRGAN model size error
2 participants