You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
this PR #2588 breaks the loading behavior in diffusers when:
accelerate.load_checkpoint_and_dispatch is used
the checkpoint contains deprecated weight names for the attention class
As you can see here, we rely on the error to rename the parameters' names on our end. Is there a better way to do this in diffusers?
The alternative is to allow us a pass a strict=True from load_checkpoint_and_dispatch- let me know what you think :)
System Info
Information
this PR #2588 breaks the loading behavior in diffusers when:
accelerate.load_checkpoint_and_dispatch
is usedAs you can see here, we rely on the error to rename the parameters' names on our end. Is there a better way to do this in diffusers?
The alternative is to allow us a pass a
strict=True
fromload_checkpoint_and_dispatch
- let me know what you think :)Reproduction
this script fails currently but it should not
The text was updated successfully, but these errors were encountered: