block swap with no_grad context #1988
kohya-ss
wants to merge
2 commits into
sd3
from
no-grad-block-swap
fix: optimize weight device swapping with no_grad context
c898e4e5
Merge branch 'sd3' into no-grad-block-swap
0e8b5d4a
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub